Merge pull request #1500 from lainedfles/config-image-generation-enabled

feat: add IMAGE_GENERATION_ENABLED env var
This commit is contained in:
Timothy Jaeryang Baek
2024-04-13 14:51:01 -07:00
committed by GitHub
2 changed files with 11 additions and 2 deletions

View File

@@ -456,5 +456,8 @@ WHISPER_MODEL_DIR = os.getenv("WHISPER_MODEL_DIR", f"{CACHE_DIR}/whisper/models"
# Images
####################################
ENABLE_IMAGE_GENERATION = (
os.environ.get("ENABLE_IMAGE_GENERATION", "").lower() == "true"
)
AUTOMATIC1111_BASE_URL = os.getenv("AUTOMATIC1111_BASE_URL", "")
COMFYUI_BASE_URL = os.getenv("COMFYUI_BASE_URL", "")