diff --git a/src/diffusers/pipelines/z_image/pipeline_z_image_omni.py b/src/diffusers/pipelines/z_image/pipeline_z_image_omni.py index 50776ceaf34d..92da2375495e 100644 --- a/src/diffusers/pipelines/z_image/pipeline_z_image_omni.py +++ b/src/diffusers/pipelines/z_image/pipeline_z_image_omni.py @@ -17,7 +17,7 @@ import PIL import torch -from transformers import AutoTokenizer, PreTrainedModel, Siglip2ImageProcessorFast, Siglip2VisionModel +from transformers import AutoTokenizer, PreTrainedModel, Siglip2ImageProcessor, Siglip2VisionModel from ...loaders import FromSingleFileMixin, ZImageLoraLoaderMixin from ...models.autoencoders import AutoencoderKL @@ -153,7 +153,7 @@ def __init__( tokenizer: AutoTokenizer, transformer: ZImageTransformer2DModel, siglip: Siglip2VisionModel, - siglip_processor: Siglip2ImageProcessorFast, + siglip_processor: Siglip2ImageProcessor, ): super().__init__()