Update model.py

2025-02-25 20:14:00 +02:00
parent d3530d31fd
commit 9ee68c2e52
1 changed files with 1 additions and 4 deletions
@@ -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(