RenjieWu commited on
Commit
4bc8bca
·
verified ·
1 Parent(s): 67cbf20

Update modeling_dots_ocr_vllm.py

Browse files
Files changed (1) hide show
  1. modeling_dots_ocr_vllm.py +11 -0
modeling_dots_ocr_vllm.py CHANGED
@@ -91,6 +91,17 @@ class DotsOCRProcessingInfo(Qwen2_5_VLProcessingInfo):
91
 
92
  return config
93
 
 
 
 
 
 
 
 
 
 
 
 
94
  def get_hf_processor(
95
  self,
96
  *,
 
91
 
92
  return config
93
 
94
+ def get_supported_mm_limits(self) -> Mapping[str, Optional[int]]:
95
+ return {"image": None, "video": 0}
96
+
97
+ def get_mm_max_tokens_per_item(
98
+ self,
99
+ seq_len: int,
100
+ mm_counts: Mapping[str, int],
101
+ ) -> Mapping[str, int]:
102
+ max_image_tokens = self.get_max_image_tokens()
103
+ return {"image": max_image_tokens, "video": 0}
104
+
105
  def get_hf_processor(
106
  self,
107
  *,