Fix: Resolve TypeError for video_processor during model loading.
#38
by
prithivMLmods
- opened
- configuration_dots.py +4 -3
configuration_dots.py
CHANGED
|
@@ -52,8 +52,8 @@ class DotsVisionConfig(PretrainedConfig):
|
|
| 52 |
|
| 53 |
class DotsOCRConfig(Qwen2Config):
|
| 54 |
model_type = "dots_ocr"
|
| 55 |
-
def __init__(self,
|
| 56 |
-
image_token_id = 151665,
|
| 57 |
video_token_id = 151656,
|
| 58 |
vision_config: Optional[dict] = None, *args, **kwargs):
|
| 59 |
super().__init__(*args, **kwargs)
|
|
@@ -67,6 +67,7 @@ class DotsOCRConfig(Qwen2Config):
|
|
| 67 |
|
| 68 |
|
| 69 |
class DotsVLProcessor(Qwen2_5_VLProcessor):
|
|
|
|
| 70 |
def __init__(self, image_processor=None, tokenizer=None, chat_template=None, **kwargs):
|
| 71 |
super().__init__(image_processor, tokenizer, chat_template=chat_template)
|
| 72 |
self.image_token = "<|imgpad|>" if not hasattr(tokenizer, "image_token") else tokenizer.image_token
|
|
@@ -74,4 +75,4 @@ class DotsVLProcessor(Qwen2_5_VLProcessor):
|
|
| 74 |
|
| 75 |
|
| 76 |
AutoProcessor.register("dots_ocr", DotsVLProcessor)
|
| 77 |
-
CONFIG_MAPPING.register("dots_ocr", DotsOCRConfig)
|
|
|
|
| 52 |
|
| 53 |
class DotsOCRConfig(Qwen2Config):
|
| 54 |
model_type = "dots_ocr"
|
| 55 |
+
def __init__(self,
|
| 56 |
+
image_token_id = 151665,
|
| 57 |
video_token_id = 151656,
|
| 58 |
vision_config: Optional[dict] = None, *args, **kwargs):
|
| 59 |
super().__init__(*args, **kwargs)
|
|
|
|
| 67 |
|
| 68 |
|
| 69 |
class DotsVLProcessor(Qwen2_5_VLProcessor):
|
| 70 |
+
attributes = ["image_processor", "tokenizer"]
|
| 71 |
def __init__(self, image_processor=None, tokenizer=None, chat_template=None, **kwargs):
|
| 72 |
super().__init__(image_processor, tokenizer, chat_template=chat_template)
|
| 73 |
self.image_token = "<|imgpad|>" if not hasattr(tokenizer, "image_token") else tokenizer.image_token
|
|
|
|
| 75 |
|
| 76 |
|
| 77 |
AutoProcessor.register("dots_ocr", DotsVLProcessor)
|
| 78 |
+
CONFIG_MAPPING.register("dots_ocr", DotsOCRConfig)
|