Add IMAGE_GENERATION_ENABLED env var

With default of `False`.
This commit is contained in:
Self Denial
2024-04-10 23:21:12 -06:00
parent 9e726a32e6
commit 3b8ea55bc8
2 changed files with 5 additions and 2 deletions

View File

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