diff --git a/Dockerfile b/Dockerfile index fa30e10..0e94af1 100644 --- a/Dockerfile +++ b/Dockerfile @@ -19,4 +19,4 @@ LABEL maintainer="zchengrong@yeah.net" \ name="trinity_aida" -CMD ["gunicorn", "-c", "gunicorn_config.py", "app.main:app" , "-e", "RABBITMQ_QUEUES=SuperResolution-dev"] \ No newline at end of file +CMD ["gunicorn", "-c", "gunicorn_config.py", "app.main:app" , "-e", "SR_RABBITMQ_QUEUES=SuperResolution" ,"-e", "GI_RABBITMQ_QUEUES=GenerateImage"] \ No newline at end of file diff --git a/app/core/config.py b/app/core/config.py index e286b5d..6bf0556 100644 --- a/app/core/config.py +++ b/app/core/config.py @@ -25,9 +25,9 @@ if DEBUG: else: LOGS_PATH = "app/logs/errors.log" -# RABBITMQ_ENV = "" -RABBITMQ_ENV = "dev" -# RABBITMQ_ENV = "local" +RABBITMQ_ENV = "" # 生产环境 +# RABBITMQ_ENV = "-dev" # 开发环境 +# RABBITMQ_ENV = "-local" # 本地测试环境 settings = Settings() @@ -61,7 +61,7 @@ SR_RABBITMQ_QUEUES = os.getenv("SR_RABBITMQ_QUEUES", "SuperResolution-local") GI_MODEL_NAME = '_stable_diffusion' GI_MODEL_URL = '10.1.1.240:7001' GI_MINIO_BUCKET = "aida-users" -GI_RABBITMQ_QUEUES = os.getenv("GI_RABBITMQ_QUEUES", f"GenerateImage-{RABBITMQ_ENV}") +GI_RABBITMQ_QUEUES = os.getenv("GI_RABBITMQ_QUEUES", f"GenerateImage{RABBITMQ_ENV}") # SEG service config SEG_MODEL_URL = '10.1.1.240:10000'