add apply_chat_template for processor
Browse files- processing_magma.py +3 -0
processing_magma.py
CHANGED
|
@@ -118,6 +118,9 @@ class MagmaProcessor(ProcessorMixin):
|
|
| 118 |
|
| 119 |
return BatchFeature(data={**text_inputs, **image_inputs})
|
| 120 |
|
|
|
|
|
|
|
|
|
|
| 121 |
# Copied from transformers.models.clip.processing_clip.CLIPProcessor.batch_decode with CLIP->Llama
|
| 122 |
def batch_decode(self, *args, **kwargs):
|
| 123 |
"""
|
|
|
|
| 118 |
|
| 119 |
return BatchFeature(data={**text_inputs, **image_inputs})
|
| 120 |
|
| 121 |
+
def apply_chat_template(self, *args, **kwargs):
|
| 122 |
+
return self.tokenizer.apply_chat_template(*args, **kwargs)
|
| 123 |
+
|
| 124 |
# Copied from transformers.models.clip.processing_clip.CLIPProcessor.batch_decode with CLIP->Llama
|
| 125 |
def batch_decode(self, *args, **kwargs):
|
| 126 |
"""
|