fix loading processor after save_pretrained with transformers 4.49+

#60
by katuni4ka - opened
Files changed (1) hide show
  1. 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