bugfix
This commit is contained in:
parent
33447d53b2
commit
68d6113070
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user