Skip to content

Commit e2ecd86

Browse files
committed
propagate changes to more models
1 parent c11ac78 commit e2ecd86

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/transformers/models/llava_next/processing_llava_next.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,8 @@ class LlavaNextProcessor(ProcessorMixin):
4040
"""
4141

4242
attributes = ["image_processor", "tokenizer"]
43-
image_processor_class = "LlavaNextImageProcessor"
44-
tokenizer_class = ("LlamaTokenizer", "LlamaTokenizerFast")
43+
image_processor_class = "AutoImageProcessor"
44+
tokenizer_class = "AutoTokenizer"
4545

4646
def __init__(self, image_processor=None, tokenizer=None):
4747
super().__init__(image_processor, tokenizer)

src/transformers/models/video_llava/processing_video_llava.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ class VideoLlavaProcessor(ProcessorMixin):
4242

4343
attributes = ["image_processor", "tokenizer"]
4444
image_processor_class = "VideoLlavaImageProcessor"
45-
tokenizer_class = ("LlamaTokenizer", "LlamaTokenizerFast")
45+
tokenizer_class = "AutoTokenizer"
4646

4747
def __init__(self, image_processor=None, tokenizer=None):
4848
super().__init__(image_processor, tokenizer)

0 commit comments

Comments
 (0)