From e2ecd865b1bbc4d85f32ad73143d68dd0a456112 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Mon, 20 May 2024 11:57:42 +0200 Subject: [PATCH] propagate changes to more models --- src/transformers/models/llava_next/processing_llava_next.py | 4 ++-- src/transformers/models/video_llava/processing_video_llava.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/transformers/models/llava_next/processing_llava_next.py b/src/transformers/models/llava_next/processing_llava_next.py index 8a4b76e9c68a..91cd544ab648 100644 --- a/src/transformers/models/llava_next/processing_llava_next.py +++ b/src/transformers/models/llava_next/processing_llava_next.py @@ -40,8 +40,8 @@ class LlavaNextProcessor(ProcessorMixin): """ attributes = ["image_processor", "tokenizer"] - image_processor_class = "LlavaNextImageProcessor" - tokenizer_class = ("LlamaTokenizer", "LlamaTokenizerFast") + image_processor_class = "AutoImageProcessor" + tokenizer_class = "AutoTokenizer" def __init__(self, image_processor=None, tokenizer=None): super().__init__(image_processor, tokenizer) diff --git a/src/transformers/models/video_llava/processing_video_llava.py b/src/transformers/models/video_llava/processing_video_llava.py index 0355d756ce27..b71600501b4c 100644 --- a/src/transformers/models/video_llava/processing_video_llava.py +++ b/src/transformers/models/video_llava/processing_video_llava.py @@ -42,7 +42,7 @@ class VideoLlavaProcessor(ProcessorMixin): attributes = ["image_processor", "tokenizer"] image_processor_class = "VideoLlavaImageProcessor" - tokenizer_class = ("LlamaTokenizer", "LlamaTokenizerFast") + tokenizer_class = "AutoTokenizer" def __init__(self, image_processor=None, tokenizer=None): super().__init__(image_processor, tokenizer)