From b5f327f350eaa8e53d8bdcaf14cd4a72307e1e8e Mon Sep 17 00:00:00 2001 From: Ye Liu Date: Thu, 6 Feb 2025 17:03:09 +0800 Subject: [PATCH] Add `Qwen2VLImageProcessorFast` into `Qwen2VLProcessor` (#35987) * Add `Qwen2VLImageProcessorFast` into `Qwen2VLProcessor` * Use `AutoImageProcessor` instead Co-authored-by: Yoni Gozlan <74535834+yonigozlan@users.noreply.github.com> --------- Co-authored-by: Yoni Gozlan <74535834+yonigozlan@users.noreply.github.com> --- src/transformers/models/qwen2_vl/processing_qwen2_vl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/qwen2_vl/processing_qwen2_vl.py b/src/transformers/models/qwen2_vl/processing_qwen2_vl.py index efeb8d80d..cd4a61d28 100644 --- a/src/transformers/models/qwen2_vl/processing_qwen2_vl.py +++ b/src/transformers/models/qwen2_vl/processing_qwen2_vl.py @@ -57,7 +57,7 @@ class Qwen2VLProcessor(ProcessorMixin): attributes = ["image_processor", "tokenizer"] valid_kwargs = ["chat_template"] - image_processor_class = "Qwen2VLImageProcessor" + image_processor_class = "AutoImageProcessor" tokenizer_class = ("Qwen2Tokenizer", "Qwen2TokenizerFast") def __init__(self, image_processor=None, tokenizer=None, chat_template=None, **kwargs):