diff --git a/app/service/comfyui_I2V/pose2v_server.py b/app/service/comfyui_I2V/pose2v_server.py index 17e3fc5..a37f246 100644 --- a/app/service/comfyui_I2V/pose2v_server.py +++ b/app/service/comfyui_I2V/pose2v_server.py @@ -13,7 +13,7 @@ from PIL import Image from minio import Minio, S3Error from moviepy.video.io.VideoFileClip import VideoFileClip -from app.core.config import settings +from app.core.config import settings, PS_RABBITMQ_QUEUES from app.schemas.comfyui_i2v import ComfyuiPose2VModel from app.service.generate_image.utils.mq import publish_status @@ -622,9 +622,9 @@ class ComfyUIServerPose2V: # 推送消息 if not settings.DEBUG: - publish_status(json.dumps(self.pose_transform_data), settings.COMFYUI_SERVER_ADDRESS) + publish_status(json.dumps(self.pose_transform_data), PS_RABBITMQ_QUEUES) logger.info( - f" [x] Sent to: {settings.COMFYUI_SERVER_ADDRESS} data:@@@@ {json.dumps(self.pose_transform_data, indent=4)}") + f" [x] Sent to: {PS_RABBITMQ_QUEUES} data:@@@@ {json.dumps(self.pose_transform_data, indent=4)}") return "\n🎉 所有任务完成!"