From 2637aad79628450654f3634572ba0c8d67271d40 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 9 Apr 2026 01:20:18 +0300
Subject: [PATCH] Fix oom fallback frame loss

---
 comfy_extras/nodes_frame_interpolation.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

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