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(