momergul commited on
Commit
ad6c176
·
verified ·
1 Parent(s): 7042921

Upload processor_flamingo.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. processor_flamingo.py +3 -4
processor_flamingo.py CHANGED
@@ -9,12 +9,11 @@ class FlamingoProcessor(ProcessorMixin):
9
  Custom processor that combines a tokenizer and feature extractor.
10
  """
11
  attributes = ["feature_extractor", "tokenizer"]
 
 
12
 
13
  def __init__(self, feature_extractor, tokenizer):
14
- self.feature_extractor = feature_extractor
15
- self.tokenizer = tokenizer
16
- self.chat_template = None
17
- #super().__init__(feature_extractor, tokenizer)
18
 
19
  def __call__(self, text=None, images=None, **kwargs):
20
  """
 
9
  Custom processor that combines a tokenizer and feature extractor.
10
  """
11
  attributes = ["feature_extractor", "tokenizer"]
12
+ feature_extractor_class = "AutoImageProcessor"
13
+ tokenizer_class = "AutoTokenizer"
14
 
15
  def __init__(self, feature_extractor, tokenizer):
16
+ super().__init__(feature_extractor, tokenizer)
 
 
 
17
 
18
  def __call__(self, text=None, images=None, **kwargs):
19
  """