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 |
*,
|