diff --git a/llmengine/qwen3_reranker.py b/llmengine/qwen3_reranker.py index 1fb9231..5e8392c 100644 --- a/llmengine/qwen3_reranker.py +++ b/llmengine/qwen3_reranker.py @@ -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]