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

Add cleanup only on the error exit (when tile < 128) before re-raising the OOM exception

Browse Source
This commit is contained in:
yash-711 2026-02-25 13:04:24 +05:30 committed by GitHub
parent 70408503fb
commit b749e0db49
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy_extras/nodes_upscale_model.py
View File

@ -89,6 +89,7 @@ class ImageUpscaleWithModel(io.ComfyNode):
except model_management.OOM_EXCEPTION as e: except model_management.OOM_EXCEPTION as e:
tile //= 2 tile //= 2
if tile < 128: if tile < 128:
upscale_model.to("cpu")
raise e raise e
# upscale_model.to("cpu") # Commented out to keep model on GPU because when processing batch images then model unnecessarily moves to CPU # upscale_model.to("cpu") # Commented out to keep model on GPU because when processing batch images then model unnecessarily moves to CPU