From 32eb6b480d32ecf7ee16ff481ec48b8479dee8bb Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 13 Oct 2025 17:47:28 +0300
Subject: [PATCH] Allow canceling mid step

---
 wanvideo/modules/model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index c2d777b..a26bb4b 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -2673,6 +2673,7 @@ class WanModel(torch.nn.Module):
                 lynx_ref_buffer = {}
 
             for b, block in enumerate(self.blocks):
+                mm.throw_exception_if_processing_interrupted()
                 block_idx = f"{b:02d}"
                 if lynx_ref_buffer is not None and not lynx_ref_feature_extractor:
                     lynx_ref_feature = lynx_ref_buffer.get(block_idx, None)