Fix: Resolve TypeError for video_processor during model loading.

#38
Files changed (1) hide show
  1. 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)