recompose the name of environment variables

jannikstdl · Apr 27, 2024 · 849a0b9 · 849a0b9
1 parent b0245a7
commit 849a0b9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/backend/apps/images/main.py b/backend/apps/images/main.py
@@ -32,13 +32,13 @@
 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 @@
     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
@@ -531,7 +531,7 @@ def create_config_file(file_path):
 # 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 @@ def create_config_file(file_path):
 
 IMAGE_STEPS = int(os.getenv("IMAGE_STEPS", 50))
 
-IMAGES_MODEL = os.getenv("IMAGES_MODEL", "")
+IMAGE_GENERATION_MODEL = os.getenv("IMAGE_GENERATION_MODEL", "")
 
 ####################################
 # Audio