ruff

Author: hlky

scripts/convert_z_image_controlnet_to_diffusers.py

@@ -1,8 +1,8 @@
 import argparse
 from contextlib import nullcontext
 
-import torch
 import safetensors.torch
+import torch
 from accelerate import init_empty_weights
 from huggingface_hub import hf_hub_download
 

src/diffusers/__init__.py

@@ -1005,8 +1005,8 @@
             WanAnimateTransformer3DModel,
             WanTransformer3DModel,
             WanVACETransformer3DModel,
-            ZImageTransformer2DModel,
             ZImageControlNetModel,
+            ZImageTransformer2DModel,
             attention_backend,
         )
         from .modular_pipelines import ComponentsManager, ComponentSpec, ModularPipeline, ModularPipelineBlocks
@@ -1359,8 +1359,8 @@
             WuerstchenCombinedPipeline,
             WuerstchenDecoderPipeline,
             WuerstchenPriorPipeline,
-            ZImagePipeline,
             ZImageControlNetPipeline,
+            ZImagePipeline,
         )
 
     try:

src/diffusers/models/controlnets/controlnet_z_image.py

@@ -438,7 +438,12 @@ def forward(
         control_context_unified = pad_sequence(control_context_unified, batch_first=True, padding_value=0.0)
         c = control_context_unified
 
-        new_kwargs = {"x": unified, "attn_mask": unified_attn_mask, "freqs_cis": unified_freqs_cis, "adaln_input": adaln_input}
+        new_kwargs = {
+            "x": unified,
+            "attn_mask": unified_attn_mask,
+            "freqs_cis": unified_freqs_cis,
+            "adaln_input": adaln_input,
+        }
 
         for layer in self.control_layers:
             if torch.is_grad_enabled() and self.gradient_checkpointing:

src/diffusers/pipelines/__init__.py

@@ -841,7 +841,7 @@
             WuerstchenDecoderPipeline,
             WuerstchenPriorPipeline,
         )
-        from .z_image import ZImagePipeline, ZImageControlNetPipeline
+        from .z_image import ZImageControlNetPipeline, ZImagePipeline
 
         try:
             if not is_onnx_available():