YohanAcxias commited on
Commit
6832ee2
·
verified ·
1 Parent(s): 9a93448

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +5 -1
handler.py CHANGED
@@ -394,7 +394,11 @@ class EndpointHandler:
394
  with open(rerank_name_path, "r", encoding="utf-8") as f:
395
  rerank_name = f.read().strip()
396
  if rerank_name:
397
- self.reranker = CrossEncoder(rerank_name, device=self.device)
 
 
 
 
398
 
399
  # -----------------------------
400
  # Input normalization / query formatting
 
394
  with open(rerank_name_path, "r", encoding="utf-8") as f:
395
  rerank_name = f.read().strip()
396
  if rerank_name:
397
+ self.reranker = CrossEncoder(
398
+ rerank_name,
399
+ max_length=512,
400
+ device=self.device,
401
+ )
402
 
403
  # -----------------------------
404
  # Input normalization / query formatting