wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-03 12:57:25 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

fix (MultiGPU): prevent freeze on manual abort when using MultiGPU CFG Split (#14235)

Browse Source 
* fix (MultiGPU): prevent freeze on manual abort when using MultiGPU CFG Split

Problem:
Upon manual abort application hangs indefinitely.
`InterruptProcessingException` inherits from `BaseException` and bypasses MultiGPU's worker error handling block so thread dies silently, leaving the main thread waiting forever for `result_q.get()`

Fix:
Catch `comfy.model_management.InterruptProcessingException` instead of `Exception` so it's caught and passed back via `result_q` to unblock the main thread when manual abort signal fires.

* oops
This commit is contained in:
Quasar of Mikus 2026-06-02 13:05:24 -04:00 committed by GitHub
parent c96fcddb81
commit e9207aa7cc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/multigpu.py
View File

@ -54,6 +54,8 @@ class MultiGPUThreadPool:
try: try:
result = fn(*args, **kwargs) result = fn(*args, **kwargs)
result_q.put((result, None)) result_q.put((result, None))
except comfy.model_management.InterruptProcessingException as e:
result_q.put((None, e))
except Exception as e: except Exception as e:
result_q.put((None, e)) result_q.put((None, e))