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
fa4d1d42
Commit
fa4d1d42
authored
Aug 08, 2024
by
Michael Poluektov
Browse files
fix: backend format test
parent
309cd645
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
2 deletions
+3
-2
backend/apps/images/utils/comfyui.py
backend/apps/images/utils/comfyui.py
+3
-2
No files found.
backend/apps/images/utils/comfyui.py
View file @
fa4d1d42
import
asyncio
import
asyncio
import
websocket
# NOTE: websocket-client (https://github.com/websocket-client/websocket-client)
import
websocket
# NOTE: websocket-client (https://github.com/websocket-client/websocket-client)
import
uuid
import
json
import
json
import
urllib.request
import
urllib.request
import
urllib.parse
import
urllib.parse
...
@@ -398,7 +397,9 @@ async def comfyui_generate_image(
...
@@ -398,7 +397,9 @@ async def comfyui_generate_image(
return
None
return
None
try
:
try
:
images
=
await
asyncio
.
to_thread
(
get_images
,
ws
,
comfyui_prompt
,
client_id
,
base_url
)
images
=
await
asyncio
.
to_thread
(
get_images
,
ws
,
comfyui_prompt
,
client_id
,
base_url
)
except
Exception
as
e
:
except
Exception
as
e
:
log
.
exception
(
f
"Error while receiving images:
{
e
}
"
)
log
.
exception
(
f
"Error while receiving images:
{
e
}
"
)
images
=
None
images
=
None
...
...
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