numb3r3 commited on
Commit
c7891eb
·
verified ·
1 Parent(s): 4de0a68

init commit

Browse files
Files changed (1) hide show
  1. modeling.py +3 -1
modeling.py CHANGED
@@ -133,7 +133,9 @@ class JinaVLForRanking(Qwen2VLForConditionalGeneration):
133
  if not hasattr(self, "_processor"):
134
  from transformers import AutoProcessor
135
 
136
- self._processor = AutoProcessor.from_pretrained(self.name_or_path, trust_remote_code=True)
 
 
137
 
138
  assert isinstance(pairs, list)
139
 
 
133
  if not hasattr(self, "_processor"):
134
  from transformers import AutoProcessor
135
 
136
+ self._processor = AutoProcessor.from_pretrained(
137
+ self.name_or_path, max_pixels=602112, min_pixels=3136, trust_remote_code=True
138
+ )
139
 
140
  assert isinstance(pairs, list)
141