Upload processor_flamingo.py with huggingface_hub
Browse files- 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 |
-
|
| 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 |
"""
|