From 9ee68c2e52bbfe59eaad240ac563aee4d95f52ca Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 25 Feb 2025 20:14:00 +0200
Subject: [PATCH] Update model.py

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

diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index 7eab064..8c2cefa 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -454,10 +454,7 @@ class WanModel(ModelMixin, ConfigMixin):
         self.main_device = main_device
         self.offload_device = offload_device
 
-        self.double_blocks_to_swap = -1
-        self.single_blocks_to_swap = -1
-        self.offload_txt_in = False
-        self.offload_img_in = False
+        self.blocks_to_swap = -1
 
         # embeddings
         self.patch_embedding = nn.Conv3d(