format

Author: susanbao

src/maxdiffusion/pipelines/wan/wan_pipeline.py

@@ -376,7 +376,7 @@ def quantize_transformer(cls, config: HyperParameters, model: WanModel, pipeline
       return model
     max_logging.log("Quantizing transformer with Qwix.")
 
-    batch_size=config.global_batch_size_to_train_on
+    batch_size = config.global_batch_size_to_train_on
     latents, prompt_embeds, timesteps = get_dummy_wan_inputs(config, pipeline, batch_size)
     model_inputs = (latents, timesteps, prompt_embeds)
     with mesh: