style

Author: tolgacangoz

src/diffusers/models/transformers/latte_transformer_3d.py

@@ -171,7 +171,9 @@ def __init__(
 
         self.gradient_checkpointing = False
 
-    def _load_from_state_dict(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs):
+    def _load_from_state_dict(
+        self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs
+    ):
         if "scale_shift_table" in state_dict:
             scale_shift_table = state_dict.pop("scale_shift_table")
             state_dict[prefix + "norm_out.linear.weight"] = scale_shift_table[1]

src/diffusers/models/transformers/pixart_transformer_2d.py

@@ -185,11 +185,13 @@ def __init__(
         )
         self.caption_projection = None
         if self.config.caption_channels is not None:
-                self.caption_projection = PixArtAlphaTextProjection(
-                    in_features=self.config.caption_channels, hidden_size=self.inner_dim
-                )
+            self.caption_projection = PixArtAlphaTextProjection(
+                in_features=self.config.caption_channels, hidden_size=self.inner_dim
+            )
 
-    def _load_from_state_dict(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs):
+    def _load_from_state_dict(
+        self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs
+    ):
         if "scale_shift_table" in state_dict:
             scale_shift_table = state_dict.pop("scale_shift_table")
             state_dict[prefix + "norm_out.linear.weight"] = scale_shift_table[1]

src/diffusers/models/transformers/transformer_allegro.py

@@ -310,7 +310,9 @@ def __init__(
 
         self.gradient_checkpointing = False
 
-    def _load_from_state_dict(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs):
+    def _load_from_state_dict(
+        self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs
+    ):
         if "scale_shift_table" in state_dict:
             scale_shift_table = state_dict.pop("scale_shift_table")
             state_dict[prefix + "norm_out.linear.weight"] = scale_shift_table[1]

src/diffusers/models/transformers/transformer_ltx.py

@@ -400,7 +400,9 @@ def __init__(
 
         self.gradient_checkpointing = False
 
-    def _load_from_state_dict(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs):
+    def _load_from_state_dict(
+        self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs
+    ):
         key = "scale_shift_table"
         if prefix + key in state_dict:
             scale_shift_table = state_dict.pop(prefix + key)

src/diffusers/models/transformers/transformer_wan.py

@@ -439,7 +439,9 @@ def __init__(
 
         self.gradient_checkpointing = False
 
-    def _load_from_state_dict(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs):
+    def _load_from_state_dict(
+        self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs
+    ):
         key = "scale_shift_table"
         if prefix + key in state_dict:
             scale_shift_table = state_dict.pop(prefix + key)

src/diffusers/models/transformers/transformer_wan_vace.py

@@ -270,7 +270,9 @@ def __init__(
 
         self.gradient_checkpointing = False
 
-    def _load_from_state_dict(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs):
+    def _load_from_state_dict(
+        self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs
+    ):
         key = "scale_shift_table"
         if prefix + key in state_dict:
             scale_shift_table = state_dict.pop(prefix + key)