This commit is contained in:
yumoqing 2025-06-21 11:29:48 +08:00
parent 33447d53b2
commit 68d6113070

View File

@ -11,5 +11,6 @@ class Qwen3Reranker(BaseReranker):
self.model = AutoModelForCausalLM.from_pretrained(model_id).eval() self.model = AutoModelForCausalLM.from_pretrained(model_id).eval()
self.model_id = model_id self.model_id = model_id
self.model_name = model_id.split('/')[-1] self.model_name = model_id.split('/')[-1]
self.max_length = 8192
llm_register('Qwen3-Reranker', Qwen3Reranker) llm_register('Qwen3-Reranker', Qwen3Reranker)