From abdac5ec3a41f88c9cd426b7f1ea5a17748d5fb7 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 14 May 2025 10:31:44 +0300
Subject: [PATCH] Update model.py

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

diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index 44c1ebd..508c76a 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -14,6 +14,7 @@ try:
     create_block_mask = torch.compile(create_block_mask)
     flex_attention = torch.compile(flex_attention)
 except:
+    BlockMask = create_block_mask = flex_attention = None
     pass
 
 from .attention import attention