|
@@ -32,13 +32,13 @@ import logging
|
|
from config import (
|
|
from config import (
|
|
SRC_LOG_LEVELS,
|
|
SRC_LOG_LEVELS,
|
|
CACHE_DIR,
|
|
CACHE_DIR,
|
|
- IMAGES_GENERATION_ENGINE,
|
|
|
|
|
|
+ IMAGE_GENERATION_ENGINE,
|
|
ENABLE_IMAGE_GENERATION,
|
|
ENABLE_IMAGE_GENERATION,
|
|
AUTOMATIC1111_BASE_URL,
|
|
AUTOMATIC1111_BASE_URL,
|
|
COMFYUI_BASE_URL,
|
|
COMFYUI_BASE_URL,
|
|
IMAGES_OPENAI_API_BASE_URL,
|
|
IMAGES_OPENAI_API_BASE_URL,
|
|
IMAGES_OPENAI_API_KEY,
|
|
IMAGES_OPENAI_API_KEY,
|
|
- IMAGES_MODEL,
|
|
|
|
|
|
+ IMAGE_GENERATION_MODEL,
|
|
IMAGE_SIZE,
|
|
IMAGE_SIZE,
|
|
IMAGE_STEPS,
|
|
IMAGE_STEPS,
|
|
)
|
|
)
|
|
@@ -59,13 +59,13 @@ app.add_middleware(
|
|
allow_headers=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
)
|
|
|
|
|
|
-app.state.ENGINE = IMAGES_GENERATION_ENGINE
|
|
|
|
|
|
+app.state.ENGINE = IMAGE_GENERATION_ENGINE
|
|
app.state.ENABLED = ENABLE_IMAGE_GENERATION
|
|
app.state.ENABLED = ENABLE_IMAGE_GENERATION
|
|
|
|
|
|
app.state.OPENAI_API_BASE_URL = IMAGES_OPENAI_API_BASE_URL
|
|
app.state.OPENAI_API_BASE_URL = IMAGES_OPENAI_API_BASE_URL
|
|
app.state.OPENAI_API_KEY = IMAGES_OPENAI_API_KEY
|
|
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
|
|
app.state.AUTOMATIC1111_BASE_URL = AUTOMATIC1111_BASE_URL
|