This commit is contained in:
yumoqing 2025-06-21 10:42:34 +08:00
parent 1ff2ee44b5
commit 85b4862405

View File

@ -8,7 +8,7 @@ class Qwen3Reranker(BaseReranker):
e = Exception(f'{model_id} is not a Qwen3-Reranker')
raise e
self.tokenizer = AutoTokenizer.from_pretrained(model_id, padding_side='left')
self.model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-Reranker-0.6B").eval()
self.model = AutoModelForCausalLM.from_pretrained(model_id).eval()
self.model_id = model_id
self.model_name = model_id.split('/')[-1]