Update modeling_dots_ocr_vllm.py
Browse files- 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 |
*,
|