recompose the name of environment variables

This commit is contained in:
Yanyutin753 2024-04-28 07:21:01 +08:00
parent b0245a7eff
commit 849a0b973d
2 changed files with 6 additions and 6 deletions

View file

@ -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