aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat generate 迁移

Browse Source
This commit is contained in:
zhouchengrong
2024-04-15 18:39:07 +08:00
parent 6739b46d40
commit e1ef5f05c8
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@@ -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"]
CMD ["gunicorn", "-c", "gunicorn_config.py", "app.main:app" , "-e", "SR_RABBITMQ_QUEUES=SuperResolution" ,"-e", "GI_RABBITMQ_QUEUES=GenerateImage"]

8
app/core/config.py
View File

@@ -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'