[PR #4448] [MERGED] fix: ComfyUI generation no longer causes FastAPI to stall for all users #60347

New Issue

GiteaMirror · 2026-05-06T03:05:24-05:00

GiteaMirror commented

2026-05-06 03:05:24 -05:00

📋 Pull Request Information

Original PR: https://github.com/open-webui/open-webui/pull/4448
Author: @JohnTheNerd
Created: 8/8/2024
Status: ✅ Merged
Merged: 8/8/2024
Merged by: @tjbck

Base: dev ← Head: comfyui-async-fix

📝 Commits (1)

958fe96 fix: ComfyUI generation no longer causes FastAPI to stall for all users

📊 Changes

2 files changed (+4 additions, -3 deletions)

View changed files

📝 backend/apps/images/main.py (+1 -1)
📝 backend/apps/images/utils/comfyui.py (+3 -2)

📄 Description

Pull Request Checklist

Note to first-time contributors: Please open a discussion post in Discussions and describe your changes before submitting a pull request.

Before submitting, make sure you've checked the following:

Target branch: Please verify that the pull request targets the dev branch.
Description: Provide a concise description of the changes made in this pull request.
Changelog: Ensure a changelog entry following the format of Keep a Changelog is added at the bottom of the PR description.
Documentation: Have you updated relevant documentation Open WebUI Docs, or other documentation sources?
Dependencies: Are there any new dependencies? Have you updated the dependency versions in the documentation?
Testing: Have you written and run sufficient tests for validating the changes?
Code review: Have you performed a self-review of your code, addressing any coding standard issues and ensuring adherence to the project's coding standards?
Prefix: To cleary categorize this pull request, prefix the pull request title, using one of the following:
- BREAKING CHANGE: Significant changes that may affect compatibility
- build: Changes that affect the build system or external dependencies
- ci: Changes to our continuous integration processes or workflows
- chore: Refactor, cleanup, or other non-functional code changes
- docs: Documentation update or addition
- feat: Introduces a new feature or enhancement to the codebase
- fix: Bug fix or error correction
- i18n: Internationalization or localization changes
- perf: Performance improvement
- refactor: Code restructuring for better maintainability, readability, or scalability
- style: Changes that do not affect the meaning of the code (white-space, formatting, missing semi-colons, etc.)
- test: Adding missing tests or correcting existing tests
- WIP: Work in progress, a temporary label for incomplete or ongoing work

Changelog Entry

Description

as the get_images() function involves a while True loop while waiting for a response from ComfyUI and is not async, when image generation is running the entire UI becomes unresponsive for all users.

furthermore, when image generation takes too long, the Docker health check starts failing.

this is certainly a bad fix as it does not convert everything to async, but rather just puts the blocking loop in a separate thread. however, it at least fixes the problem for now.

Added

n/a

Changed

started running ComfyUI's image generation in a separate thread, instead of blocking the FastAPI thread

Deprecated

n/a

Removed

n/a

Fixed

the UI no longer becomes fully inaccessible for all users while image generation runs
Open WebUI no longer crashes when running an image generation request that takes longer than the health check timeout

Security

n/a

Breaking Changes

n/a

Additional Information

simultaneous image generation requests seem to fail after this PR. in contrast, it's way better to fail gracefully than simply hang or crash
this is certainly an ugly fix that I put together just to stop my server from continuously crashing (I share my instance with my friends, and since many people liked Flux which itself is a massive model, my server was down more often than it was up).
a proper fix involves running the ComfyUI stuff in a separate thread, queuing things inside Open-WebUI itself. however, I don't really have time to look at it right now and it's quite bad UX to have the UI hang/crash, so here's an upstream PR until I get to a better fix.

Screenshots or Videos

n/a

_{🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.}

## 📋 Pull Request Information **Original PR:** https://github.com/open-webui/open-webui/pull/4448 **Author:** [@JohnTheNerd](https://github.com/JohnTheNerd) **Created:** 8/8/2024 **Status:** ✅ Merged **Merged:** 8/8/2024 **Merged by:** [@tjbck](https://github.com/tjbck) **Base:** `dev` ← **Head:** `comfyui-async-fix` --- ### 📝 Commits (1) - [`958fe96`](https://github.com/open-webui/open-webui/commit/958fe9639a08732c04b102ca494d11f522b33b34) fix: ComfyUI generation no longer causes FastAPI to stall for all users ### 📊 Changes **2 files changed** (+4 additions, -3 deletions) <details> <summary>View changed files</summary> 📝 `backend/apps/images/main.py` (+1 -1) 📝 `backend/apps/images/utils/comfyui.py` (+3 -2) </details> ### 📄 Description # Pull Request Checklist ### Note to first-time contributors: Please open a discussion post in [Discussions](https://github.com/open-webui/open-webui/discussions) and describe your changes before submitting a pull request. **Before submitting, make sure you've checked the following:** - [x] **Target branch:** Please verify that the pull request targets the `dev` branch. - [x] **Description:** Provide a concise description of the changes made in this pull request. - [x] **Changelog:** Ensure a changelog entry following the format of [Keep a Changelog](https://keepachangelog.com/) is added at the bottom of the PR description. - [x] **Documentation:** Have you updated relevant documentation [Open WebUI Docs](https://github.com/open-webui/docs), or other documentation sources? - [x] **Dependencies:** Are there any new dependencies? Have you updated the dependency versions in the documentation? - [x] **Testing:** Have you written and run sufficient tests for validating the changes? - [x] **Code review:** Have you performed a self-review of your code, addressing any coding standard issues and ensuring adherence to the project's coding standards? - [x] **Prefix:** To cleary categorize this pull request, prefix the pull request title, using one of the following: - **BREAKING CHANGE**: Significant changes that may affect compatibility - **build**: Changes that affect the build system or external dependencies - **ci**: Changes to our continuous integration processes or workflows - **chore**: Refactor, cleanup, or other non-functional code changes - **docs**: Documentation update or addition - **feat**: Introduces a new feature or enhancement to the codebase - **fix**: Bug fix or error correction - **i18n**: Internationalization or localization changes - **perf**: Performance improvement - **refactor**: Code restructuring for better maintainability, readability, or scalability - **style**: Changes that do not affect the meaning of the code (white-space, formatting, missing semi-colons, etc.) - **test**: Adding missing tests or correcting existing tests - **WIP**: Work in progress, a temporary label for incomplete or ongoing work # Changelog Entry ### Description as the get_images() function involves a `while True` loop while waiting for a response from ComfyUI and is not async, when image generation is running the entire UI becomes unresponsive for all users. furthermore, when image generation takes too long, the Docker health check starts failing. this is certainly a bad fix as it does not convert everything to async, but rather just puts the blocking loop in a separate thread. however, it at least fixes the problem for now. ### Added n/a ### Changed started running ComfyUI's image generation in a separate thread, instead of blocking the FastAPI thread ### Deprecated n/a ### Removed n/a ### Fixed - the UI no longer becomes fully inaccessible for all users while image generation runs - Open WebUI no longer crashes when running an image generation request that takes longer than the health check timeout ### Security n/a ### Breaking Changes n/a --- ### Additional Information - simultaneous image generation requests seem to fail after this PR. in contrast, it's way better to fail gracefully than simply hang or crash - this is certainly an ugly fix that I put together just to stop my server from continuously crashing (I share my instance with my friends, and since many people liked Flux which itself is a massive model, my server was down more often than it was up). - a proper fix involves running the ComfyUI stuff in a separate thread, queuing things inside Open-WebUI itself. however, I don't really have time to look at it right now and it's quite bad UX to have the UI hang/crash, so here's an upstream PR until I get to a better fix. ### Screenshots or Videos n/a --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>

GiteaMirror added the pull-request label 2026-05-06 03:05:24 -05:00

GiteaMirror closed this issue

2026-05-06 03:05:26 -05:00

Sign in to join this conversation.

Branches Tags

1 Participants

Notifications

Due Date

No due date set.

Dependencies

No dependencies set.

Reference: github-starred/open-webui#60347