numb3r3 commited on
Commit
c3b9a22
·
verified ·
1 Parent(s): f084474

fix model length

Browse files
Files changed (2) hide show
  1. README.md +2 -0
  2. modeling.py +2 -2
README.md CHANGED
@@ -241,6 +241,8 @@ We also evaluate the performance of the `jina-reranker-m0` across four text-to-t
241
  For complete benchmark results, please refer to the [online results table](https://docs.google.com/spreadsheets/d/1KrCD7l0lhzMkyg3z-gEDmymxe4Eun9Z-C0kU3_cxw7Q/edit?usp=sharing).
242
 
243
 
 
 
244
  # Contact
245
 
246
  Join our [Discord community](https://discord.jina.ai/) and chat with other community members about ideas.
 
241
  For complete benchmark results, please refer to the [online results table](https://docs.google.com/spreadsheets/d/1KrCD7l0lhzMkyg3z-gEDmymxe4Eun9Z-C0kU3_cxw7Q/edit?usp=sharing).
242
 
243
 
244
+
245
+
246
  # Contact
247
 
248
  Join our [Discord community](https://discord.jina.ai/) and chat with other community members about ideas.
modeling.py CHANGED
@@ -122,8 +122,8 @@ class JinaVLForRanking(Qwen2VLForConditionalGeneration):
122
  self,
123
  pairs: Union[List[Tuple[str, str]], Tuple[str, str]],
124
  batch_size: int = 8,
125
- max_length: int = 8192,
126
- max_query_length: int = 512,
127
  max_doc_length: Optional[int] = None,
128
  query_type: str = 'text',
129
  doc_type: str = 'text',
 
122
  self,
123
  pairs: Union[List[Tuple[str, str]], Tuple[str, str]],
124
  batch_size: int = 8,
125
+ max_length: int = 10240,
126
+ max_query_length: int = 1024,
127
  max_doc_length: Optional[int] = None,
128
  query_type: str = 'text',
129
  doc_type: str = 'text',