Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
open-webui
Commits
849a0b97
Commit
849a0b97
authored
Apr 28, 2024
by
Yanyutin753
Browse files
recompose the name of environment variables
parent
b0245a7e
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
6 additions
and
6 deletions
+6
-6
backend/apps/images/main.py
backend/apps/images/main.py
+4
-4
backend/config.py
backend/config.py
+2
-2
No files found.
backend/apps/images/main.py
View file @
849a0b97
...
@@ -32,13 +32,13 @@ import logging
...
@@ -32,13 +32,13 @@ import logging
from
config
import
(
from
config
import
(
SRC_LOG_LEVELS
,
SRC_LOG_LEVELS
,
CACHE_DIR
,
CACHE_DIR
,
IMAGE
S
_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
,
IMAGE
S
_MODEL
,
IMAGE
_GENERATION
_MODEL
,
IMAGE_SIZE
,
IMAGE_SIZE
,
IMAGE_STEPS
,
IMAGE_STEPS
,
)
)
...
@@ -59,13 +59,13 @@ app.add_middleware(
...
@@ -59,13 +59,13 @@ app.add_middleware(
allow_headers
=
[
"*"
],
allow_headers
=
[
"*"
],
)
)
app
.
state
.
ENGINE
=
IMAGE
S
_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
=
IMAGE
S
_MODEL
app
.
state
.
MODEL
=
IMAGE
_GENERATION
_MODEL
app
.
state
.
AUTOMATIC1111_BASE_URL
=
AUTOMATIC1111_BASE_URL
app
.
state
.
AUTOMATIC1111_BASE_URL
=
AUTOMATIC1111_BASE_URL
...
...
backend/config.py
View file @
849a0b97
...
@@ -531,7 +531,7 @@ WHISPER_MODEL_AUTO_UPDATE = (
...
@@ -531,7 +531,7 @@ WHISPER_MODEL_AUTO_UPDATE = (
# Images
# Images
####################################
####################################
IMAGE
S
_GENERATION_ENGINE
=
os
.
getenv
(
"IMAGE
S
_GENERATION_ENGINE"
,
""
)
IMAGE_GENERATION_ENGINE
=
os
.
getenv
(
"IMAGE_GENERATION_ENGINE"
,
""
)
ENABLE_IMAGE_GENERATION
=
(
ENABLE_IMAGE_GENERATION
=
(
os
.
environ
.
get
(
"ENABLE_IMAGE_GENERATION"
,
""
).
lower
()
==
"true"
os
.
environ
.
get
(
"ENABLE_IMAGE_GENERATION"
,
""
).
lower
()
==
"true"
...
@@ -549,7 +549,7 @@ IMAGE_SIZE = os.getenv("IMAGE_SIZE", "512x512")
...
@@ -549,7 +549,7 @@ IMAGE_SIZE = os.getenv("IMAGE_SIZE", "512x512")
IMAGE_STEPS
=
int
(
os
.
getenv
(
"IMAGE_STEPS"
,
50
))
IMAGE_STEPS
=
int
(
os
.
getenv
(
"IMAGE_STEPS"
,
50
))
IMAGE
S
_MODEL
=
os
.
getenv
(
"IMAGE
S
_MODEL"
,
""
)
IMAGE
_GENERATION
_MODEL
=
os
.
getenv
(
"IMAGE
_GENERATION
_MODEL"
,
""
)
####################################
####################################
# Audio
# Audio
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment