Update modelling_magiv2.py

modelling_magiv2.py

@@ -413,7 +413,7 @@ class Magiv2Model(PreTrainedModel):
 
         return crop_embeddings_for_batch
     
-    def predict_ocr(self, images, crop_bboxes, move_to_device_fn=None, use_tqdm=False, batch_size=32, max_new_tokens=512):
+    def predict_ocr(self, images, crop_bboxes, move_to_device_fn=None, use_tqdm=False, batch_size=32, max_new_tokens=64):
         assert not self.config.disable_ocr
         move_to_device_fn = self.move_to_device if move_to_device_fn is None else move_to_device_fn