tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

Add IMAGE_GENERATION_ENABLED env var

Browse source 
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
Download patch file Download diff file Expand all files Collapse all files

4
backend/apps/images/main.py
View file

@ -29,7 +29,7 @@ import base64
import json
import logging
from config import SRC_LOG_LEVELS, CACHE_DIR, AUTOMATIC1111_BASE_URL, COMFYUI_BASE_URL
from config import SRC_LOG_LEVELS, CACHE_DIR, IMAGE_GENERATION_ENABLED, AUTOMATIC1111_BASE_URL, COMFYUI_BASE_URL
log = logging.getLogger(__name__)
@ -48,7 +48,7 @@ app.add_middleware(
)
app.state.ENGINE = ""
app.state.ENABLED = False
app.state.ENABLED = IMAGE_GENERATION_ENABLED
app.state.OPENAI_API_KEY = ""
app.state.MODEL = ""