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-05-09 17:56:51 +08:00
parent d59ce3f437
commit 36677b1a86
1 changed files with 41 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
app/core/config.py
View File

@@ -54,6 +54,13 @@ RABBITMQ_PARAMS = {
"virtual_host": "/" "virtual_host": "/"
} }
# milvus 配置
MILVUS_DB_HOST = "10.1.1.240"
MILVUS_ALIAS = "default"
MILVUS_PORT = "19530"
MILVUS_TABLE_KEYPOINT = "keypoint_cache"
MILVUS_TABLE_SEG = "seg_cache"
# attribute service config # attribute service config
ATT_TRITON_URL = "10.1.1.240:8020" ATT_TRITON_URL = "10.1.1.240:8020"
@@ -65,7 +72,7 @@ SR_RABBITMQ_QUEUES = os.getenv("SR_RABBITMQ_QUEUES", f"SuperResolution{RABBITMQ_
# GenerateImage service config # GenerateImage service config
GI_MODEL_NAME = 'stable_diffusion_xl' GI_MODEL_NAME = 'stable_diffusion_xl'
GI_MODEL_URL = '10.1.1.150:8001' GI_MODEL_URL = '10.1.1.240:10041'
GI_MINIO_BUCKET = "aida-users" 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}")
GI_SYS_IMAGE_URL = "aida-sys-image/generate_image/white_image.jpg" GI_SYS_IMAGE_URL = "aida-sys-image/generate_image/white_image.jpg"
@@ -77,3 +84,36 @@ SEGMENTATION = {
"input": "seg_input__0", "input": "seg_input__0",
"output": "seg_output__0", "output": "seg_output__0",
} }
# DESIGN config
DESIGN_MODEL_URL = '10.1.1.240:9000'
AIDA_CLOTHING = "aida-clothing"
# 优先级
PRIORITY_DICT = {
'earring_front': 99,
'bag_front': 98,
'hairstyle_front': 97,
'outwear_front': 20,
'tops_front': 19,
'dress_front': 18,
'blouse_front': 17,
'skirt_front': 16,
'trousers_front': 15,
'bottoms_front': 14,
'shoes_right': 1,
'shoes_left': 1,
'body': 0,
'bottoms_back': -14,
'trousers_back': -15,
'skirt_back': -16,
'blouse_back': -17,
'dress_back': -18,
'tops_back': -19,
'outwear_back': -20,
'hairstyle_back': -97,
'bag_back': -98,
'earring_back': -99,
}