mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-29 07:22:12 +03:00
update
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user