feat 产品图打光模型部署

fix
2024-06-19 16:44:04 +08:00
parent 8476bb3727
commit d04c3857fc
3 changed files with 41 additions and 81 deletions
--- a/app/core/config.py
+++ b/app/core/config.py
@@ -124,7 +124,7 @@ GPI_MODEL_URL = '10.1.1.240:10061'

 # Generate Single Logo service config
 GRI_RABBITMQ_QUEUES = os.getenv("GEN_PRODUCT_IMAGE_RABBITMQ_QUEUES", f"Relight{RABBITMQ_ENV}")
-GRI_MODEL_NAME = 'stable_diffusion_1_5'
+GRI_MODEL_NAME = 'diffusion_relight_ensemble'
 GRI_MODEL_URL = '10.1.1.150:8001'

 # SEG service config