diff --git a/app/core/config.py b/app/core/config.py index 3309d85..03be10a 100644 --- a/app/core/config.py +++ b/app/core/config.py @@ -36,7 +36,6 @@ RABBITMQ_ENV = "-prod" # 生产环境 JAVA_STREAM_API_URL = os.getenv("JAVA_STREAM_API_URL", "https://api.aida.com.hk/api/third/party/receiveDesignResults") - settings = Settings() # minio 配置 @@ -100,7 +99,7 @@ OPENAI_MODEL_LIST = {"gpt-3.5-turbo-0613", SR_MODEL_NAME = "super_resolution" SR_TRITON_URL = "10.1.1.240:10031" SR_MINIO_BUCKET = "aida-users" -SR_RABBITMQ_QUEUES = os.getenv("SR_RABBITMQ_QUEUES", f"SuperResolution{RABBITMQ_ENV}") +SR_RABBITMQ_QUEUES = f"SuperResolution{RABBITMQ_ENV}" # GenerateImage service config FAST_GI_MODEL_URL = '10.1.1.243:10011' @@ -112,21 +111,20 @@ GI_MODEL_NAME = 'flux' GMV_MODEL_URL = '10.1.1.243:10081' GMV_MODEL_NAME = 'multi_view' -GMV_RABBITMQ_QUEUES = os.getenv("GMV_RABBITMQ_QUEUES", f"GenerateMultiView{RABBITMQ_ENV}") - +GMV_RABBITMQ_QUEUES = f"GenerateMultiView{RABBITMQ_ENV}" GI_MINIO_BUCKET = "aida-users" -GI_RABBITMQ_QUEUES = os.getenv("GI_RABBITMQ_QUEUES", f"GenerateImage{RABBITMQ_ENV}") +GI_RABBITMQ_QUEUES = f"GenerateImage{RABBITMQ_ENV}" GI_SYS_IMAGE_URL = "aida-sys-image/generate_image/white_image.jpg" # SLOGAN service config -SLOGAN_RABBITMQ_QUEUES = os.getenv("SLOGAN_RABBITMQ_QUEUES", f"Slogan{RABBITMQ_ENV}") +SLOGAN_RABBITMQ_QUEUES = f"Slogan{RABBITMQ_ENV}" # Generate Single Logo service config GSL_MODEL_URL = '10.1.1.243:10041' GSL_MINIO_BUCKET = "aida-users" GSL_MODEL_NAME = 'stable_diffusion_xl_transparent' -GEN_SINGLE_LOGO_RABBITMQ_QUEUES = os.getenv("GEN_SINGLE_LOGO_RABBITMQ_QUEUES", f"GenSingleLogo{RABBITMQ_ENV}") +GEN_SINGLE_LOGO_RABBITMQ_QUEUES = f"GenSingleLogo{RABBITMQ_ENV}" # Generate Product service config # GPI_RABBITMQ_QUEUES = os.getenv("GEN_PRODUCT_IMAGE_RABBITMQ_QUEUES", f"ToProductImage{RABBITMQ_ENV}") @@ -134,13 +132,13 @@ GEN_SINGLE_LOGO_RABBITMQ_QUEUES = os.getenv("GEN_SINGLE_LOGO_RABBITMQ_QUEUES", f # GPI_MODEL_URL = '10.1.1.243:10051' # Generate Product service config 旧版product img 模型 -GPI_RABBITMQ_QUEUES = os.getenv("GEN_PRODUCT_IMAGE_RABBITMQ_QUEUES", f"ToProductImage{RABBITMQ_ENV}") +GPI_RABBITMQ_QUEUES = f"ToProductImage{RABBITMQ_ENV}" GPI_MODEL_NAME_OVERALL = 'diffusion_ensemble_all' GPI_MODEL_NAME_SINGLE = 'stable_diffusion_1_5_cnet' GPI_MODEL_URL = '10.1.1.243:10051' # Generate Single Logo service config -GRI_RABBITMQ_QUEUES = os.getenv("GEN_RELIGHT_IMAGE_RABBITMQ_QUEUES", f"Relight{RABBITMQ_ENV}") +GRI_RABBITMQ_QUEUES = f"Relight{RABBITMQ_ENV}" GRI_MODEL_NAME_OVERALL = 'diffusion_relight_ensemble' GRI_MODEL_NAME_SINGLE = 'stable_diffusion_1_5_relight' GRI_MODEL_URL = '10.1.1.240:10051'