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

feat

Browse Source 
fix  relight 新增single item模式
This commit is contained in:
zhouchengrong
2024-07-05 15:45:48 +08:00
parent bc1c903d38
commit 9d0689d98e
4 changed files with 32 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/core/config.py
View File

@@ -127,7 +127,8 @@ GPI_MODEL_URL = '10.1.1.240:10041'
# Generate Single Logo service config
GRI_RABBITMQ_QUEUES = os.getenv("GEN_RELIGHT_IMAGE_RABBITMQ_QUEUES", f"Relight{RABBITMQ_ENV}")
GRI_MODEL_NAME = 'diffusion_relight_ensemble'
GRI_MODEL_NAME_OVERALL = 'diffusion_relight_ensemble'
GRI_MODEL_NAME_SINGLE = 'stable_diffusion_1_5_relight'
GRI_MODEL_URL = '10.1.1.240:10051'
# SEG service config