From e1ef5f05c8ed6e96285562c89d7d97f150c752c5 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Mon, 15 Apr 2024 18:39:07 +0800 Subject: [PATCH] =?UTF-8?q?feat=20generate=20=E8=BF=81=E7=A7=BB?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Dockerfile | 2 +- app/core/config.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) 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'