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')
|
e = Exception(f'{model_id} is not a Qwen3-Reranker')
|
||||||
raise e
|
raise e
|
||||||
self.tokenizer = AutoTokenizer.from_pretrained(model_id, padding_side='left')
|
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_id = model_id
|
||||||
self.model_name = model_id.split('/')[-1]
|
self.model_name = model_id.split('/')[-1]
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user