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

Fix oom fallback frame loss

Browse Source
This commit is contained in:
kijai 2026-04-09 01:20:18 +03:00
parent 36a9a60368
commit 2637aad796
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy_extras/nodes_frame_interpolation.py
View File

@ -166,6 +166,7 @@ class FrameInterpolate(io.ComfyNode):
feat_cache["img1"] = inference_model.extract_features(img1_single) feat_cache["img1"] = inference_model.extract_features(img1_single)
feat_cache["next"] = feat_cache["img1"] feat_cache["next"] = feat_cache["img1"]
used_multi = False
if multi_fn is not None: if multi_fn is not None:
# Models with timestep-independent flow can compute it once for all timesteps # Models with timestep-independent flow can compute it once for all timesteps
try: try:
@ -174,12 +175,12 @@ class FrameInterpolate(io.ComfyNode):
out_idx += num_interp out_idx += num_interp
pbar.update(num_interp) pbar.update(num_interp)
tqdm_bar.update(num_interp) tqdm_bar.update(num_interp)
used_multi = True
except model_management.OOM_EXCEPTION: except model_management.OOM_EXCEPTION:
# Fall back to single-timestep calls
model_management.soft_empty_cache() model_management.soft_empty_cache()
multi_fn = None multi_fn = None # fall through to single-timestep path
continue
else: if not used_multi:
j = 0 j = 0
while j < num_interp: while j < num_interp:
b = min(batch, num_interp - j) b = min(batch, num_interp - j)