diff --git a/backend/apps/images/main.py b/backend/apps/images/main.py index 3b80a41b3..fd72b203c 100644 --- a/backend/apps/images/main.py +++ b/backend/apps/images/main.py @@ -32,13 +32,13 @@ import logging from config import ( SRC_LOG_LEVELS, CACHE_DIR, - IMAGES_GENERATION_ENGINE, + IMAGE_GENERATION_ENGINE, ENABLE_IMAGE_GENERATION, AUTOMATIC1111_BASE_URL, COMFYUI_BASE_URL, IMAGES_OPENAI_API_BASE_URL, IMAGES_OPENAI_API_KEY, - IMAGES_MODEL, + IMAGE_GENERATION_MODEL, IMAGE_SIZE, IMAGE_STEPS, ) @@ -59,13 +59,13 @@ app.add_middleware( allow_headers=["*"], ) -app.state.ENGINE = IMAGES_GENERATION_ENGINE +app.state.ENGINE = IMAGE_GENERATION_ENGINE app.state.ENABLED = ENABLE_IMAGE_GENERATION app.state.OPENAI_API_BASE_URL = IMAGES_OPENAI_API_BASE_URL app.state.OPENAI_API_KEY = IMAGES_OPENAI_API_KEY -app.state.MODEL = IMAGES_MODEL +app.state.MODEL = IMAGE_GENERATION_MODEL app.state.AUTOMATIC1111_BASE_URL = AUTOMATIC1111_BASE_URL diff --git a/backend/config.py b/backend/config.py index b05f5cee9..f864062d9 100644 --- a/backend/config.py +++ b/backend/config.py @@ -531,7 +531,7 @@ WHISPER_MODEL_AUTO_UPDATE = ( # Images #################################### -IMAGES_GENERATION_ENGINE = os.getenv("IMAGES_GENERATION_ENGINE", "") +IMAGE_GENERATION_ENGINE = os.getenv("IMAGE_GENERATION_ENGINE", "") ENABLE_IMAGE_GENERATION = ( os.environ.get("ENABLE_IMAGE_GENERATION", "").lower() == "true" @@ -549,7 +549,7 @@ IMAGE_SIZE = os.getenv("IMAGE_SIZE", "512x512") IMAGE_STEPS = int(os.getenv("IMAGE_STEPS", 50)) -IMAGES_MODEL = os.getenv("IMAGES_MODEL", "") +IMAGE_GENERATION_MODEL = os.getenv("IMAGE_GENERATION_MODEL", "") #################################### # Audio