diff --git a/app/core/config.py b/app/core/config.py index 6d5db20..cccabb9 100644 --- a/app/core/config.py +++ b/app/core/config.py @@ -54,6 +54,13 @@ RABBITMQ_PARAMS = { "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 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 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_RABBITMQ_QUEUES = os.getenv("GI_RABBITMQ_QUEUES", f"GenerateImage{RABBITMQ_ENV}") GI_SYS_IMAGE_URL = "aida-sys-image/generate_image/white_image.jpg" @@ -77,3 +84,36 @@ SEGMENTATION = { "input": "seg_input__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, +} +