style

Author: tolgacangoz

scripts/convert_wan_to_diffusers.py

@@ -9,14 +9,12 @@
 from transformers import (
     AutoProcessor,
     AutoTokenizer,
-    CLIPVisionModelWithProjection,
-    UMT5EncoderModel,
-    Wav2Vec2ForCTC,
-    Wav2Vec2Processor,
     CLIPImageProcessor,
     CLIPVisionModel,
     CLIPVisionModelWithProjection,
     UMT5EncoderModel,
+    Wav2Vec2ForCTC,
+    Wav2Vec2Processor,
 )
 
 from diffusers import (
@@ -223,6 +221,7 @@
     },
 }
 
+
 # TODO: Verify this and simplify if possible.
 def convert_animate_motion_encoder_weights(key: str, state_dict: Dict[str, Any], final_conv_idx: int = 8) -> None:
     """

src/diffusers/__init__.py

@@ -268,8 +268,8 @@
             "UNetSpatioTemporalConditionModel",
             "UVit2DModel",
             "VQModel",
-            "WanS2VTransformer3DModel",
             "WanAnimateTransformer3DModel",
+            "WanS2VTransformer3DModel",
             "WanTransformer3DModel",
             "WanVACETransformer3DModel",
             "attention_backend",

src/diffusers/models/__init__.py

@@ -108,8 +108,8 @@
     _import_structure["transformers.transformer_skyreels_v2"] = ["SkyReelsV2Transformer3DModel"]
     _import_structure["transformers.transformer_temporal"] = ["TransformerTemporalModel"]
     _import_structure["transformers.transformer_wan"] = ["WanTransformer3DModel"]
-    _import_structure["transformers.transformer_wan_s2v"] = ["WanS2VTransformer3DModel"]
     _import_structure["transformers.transformer_wan_animate"] = ["WanAnimateTransformer3DModel"]
+    _import_structure["transformers.transformer_wan_s2v"] = ["WanS2VTransformer3DModel"]
     _import_structure["transformers.transformer_wan_vace"] = ["WanVACETransformer3DModel"]
     _import_structure["unets.unet_1d"] = ["UNet1DModel"]
     _import_structure["unets.unet_2d"] = ["UNet2DModel"]

tests/quantization/gguf/test_gguf.py

@@ -749,6 +749,7 @@ def get_dummy_inputs(self):
             "timestep": torch.tensor([1]).to(torch_device, self.torch_dtype),
         }
 
+
 class WanAnimateGGUFSingleFileTests(GGUFSingleFileTesterMixin, unittest.TestCase):
     ckpt_path = "https://huggingface.co/QuantStack/Wan2.2-Animate-14B-GGUF/blob/main/Wan2.2-Animate-14B-Q3_K_S.gguf"
     torch_dtype = torch.bfloat16