jiang-cc commited on
Commit
6f83236
·
verified ·
1 Parent(s): 71f54e9

Upload processor

Browse files
Files changed (1) hide show
  1. modeling_yangjian.py +1 -1
modeling_yangjian.py CHANGED
@@ -31,7 +31,7 @@ class YangJianConfig(Qwen2_5_VLConfig):
31
  self.architectures = ["YangJianVLForConditionalGeneration"]
32
 
33
  class YangJianProcessor(Qwen2_5_VLProcessor):
34
-
35
  def __init__(self, image_processor=None, tokenizer=None, video_processor=None, chat_template=None, **kwargs):
36
  super().__init__(image_processor, tokenizer, video_processor, chat_template, **kwargs)
37
  self.compare_token_size = 100 if "compare_token_size" not in kwargs else kwargs["compare_token_size"]
 
31
  self.architectures = ["YangJianVLForConditionalGeneration"]
32
 
33
  class YangJianProcessor(Qwen2_5_VLProcessor):
34
+ config_class = YangJianConfig
35
  def __init__(self, image_processor=None, tokenizer=None, video_processor=None, chat_template=None, **kwargs):
36
  super().__init__(image_processor, tokenizer, video_processor, chat_template, **kwargs)
37
  self.compare_token_size = 100 if "compare_token_size" not in kwargs else kwargs["compare_token_size"]