Upload processor
Browse files- 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"]
|