wan2-1-fast-radial-attn-2

Running on Zero

multimodalart HF Staff commited on Jul 2

Commit

bbaad05

verified ·

1 Parent(s): e589a0c

Update radial_attn/models/wan/sparse_transformer.py

Files changed (1) hide show

radial_attn/models/wan/sparse_transformer.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 from diffusers.models.transformers.transformer_wan import WanTransformerBlock, WanTransformer3DModel
 from diffusers import WanPipeline, WanImageToVideoPipeline
-    from diffusers.image_processor import PipelineImageInput
 from diffusers.models.modeling_outputs import Transformer2DModelOutput
 from diffusers.utils import USE_PEFT_BACKEND, logging, scale_lora_layers, unscale_lora_layers
 logger = logging.get_logger(__name__)

 from diffusers.models.transformers.transformer_wan import WanTransformerBlock, WanTransformer3DModel
 from diffusers import WanPipeline, WanImageToVideoPipeline
+from diffusers.image_processor import PipelineImageInput
 from diffusers.models.modeling_outputs import Transformer2DModelOutput
 from diffusers.utils import USE_PEFT_BACKEND, logging, scale_lora_layers, unscale_lora_layers
 logger = logging.get_logger(__name__)