jw2yang commited on
Commit
fb92d9b
·
1 Parent(s): 6bbe6fa

add apply_chat_template for processor

Browse files
Files changed (1) hide show
  1. 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
  """