diff --git a/src/diffusers/modular_pipelines/wan/encoders.py b/src/diffusers/modular_pipelines/wan/encoders.py index c60fb150bf..d25e96414d 100644 --- a/src/diffusers/modular_pipelines/wan/encoders.py +++ b/src/diffusers/modular_pipelines/wan/encoders.py @@ -314,7 +314,7 @@ class WanImageEncoderStep(PipelineBlock): @staticmethod def encode_image( components, - image: PipelineImageInput, + image: Union[PipelineImageInput, List[PipelineImageInput]], device: torch.device, ): image = components.image_processor(images=image, return_tensors="pt").to(device)