fix loading processor after save_pretrained with transformers 4.49+
#60
by
katuni4ka
- opened
- processing_phi4mm.py +1 -1
processing_phi4mm.py
CHANGED
@@ -506,7 +506,7 @@ class Phi4MMProcessor(ProcessorMixin):
|
|
506 |
image_processor_class = "AutoImageProcessor" # Phi4MMImageProcessor will be registered later
|
507 |
audio_processor_class = "AutoFeatureExtractor" # Phi4MMAudioFeatureExtractor will be registered later
|
508 |
|
509 |
-
def __init__(self, image_processor, audio_processor, tokenizer):
|
510 |
self.image_processor = image_processor
|
511 |
self.audio_processor = audio_processor
|
512 |
self.tokenizer = tokenizer
|
|
|
506 |
image_processor_class = "AutoImageProcessor" # Phi4MMImageProcessor will be registered later
|
507 |
audio_processor_class = "AutoFeatureExtractor" # Phi4MMAudioFeatureExtractor will be registered later
|
508 |
|
509 |
+
def __init__(self, image_processor, audio_processor, tokenizer, **kwargs):
|
510 |
self.image_processor = image_processor
|
511 |
self.audio_processor = audio_processor
|
512 |
self.tokenizer = tokenizer
|