xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-14 08:43:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(comfy_extras): Fix OOM Tile

Browse Source
This commit is contained in:
MagicBear 2025-07-04 02:34:01 +08:00
parent 3a3b5babf2
commit ddec938cde
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy_extras/nodes_upscale_model.py
View File

@ -87,14 +87,18 @@ class ImageUpscaleWithModel:
pbar = comfy.utils.ProgressBar(steps)
s = comfy.utils.tiled_scale(in_img, parallel_model, tile_x=tile, tile_y=tile, overlap=overlap, upscale_amount=upscale_model.scale, pbar=pbar, output_device=device)
oom = False
except torch.OutOfMemoryError as e:
tile //= 2
if tile < 128:
raise e
except model_management.OOM_EXCEPTION as e:
tile //= 2
if tile < 128:
raise e
upscale_model.to("cpu")
s = torch.clamp(s.movedim(-3,-1), min=0, max=1.0)
s.to("cpu")
s = torch.clamp(s.movedim(-3,-1), min=0, max=1.0)
return (s,)
NODE_CLASS_MAPPINGS = {