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