bugfix
This commit is contained in:
parent
1ff2ee44b5
commit
85b4862405
@ -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]
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user