update reranker score

545999961 · 545999961 · commit e37bc46137ae · 2024-11-06T22:50:28.000+08:00
diff --git a/scripts/add_reranker_score.py b/scripts/add_reranker_score.py
@@ -29,7 +29,7 @@ class ModelArgs:
         default=None, metadata={"help": "The reranker name or path."}
     )
     reranker_model_class: Optional[str] = field(
-        default="auto", metadata={"help": "The reranker model class. Available classes: ['auto', 'encoder-only-base', 'decoder-only-base', 'decoder-only-layerwise', 'decoder-only-lightweight']. Default: auto.", "choices": ["auto", "encoder-only-base", "decoder-only-base", "decoder-only-layerwise", "decoder-only-lightweight"]}
+        default=None, metadata={"help": "The reranker model class. Available classes: ['auto', 'encoder-only-base', 'decoder-only-base', 'decoder-only-layerwise', 'decoder-only-lightweight']. Default: auto.", "choices": ["encoder-only-base", "decoder-only-base", "decoder-only-layerwise", "decoder-only-lightweight"]}
     )
     reranker_peft_path: Optional[str] = field(
         default=None, metadata={"help": "The reranker peft path."}

Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,7 @@ class ModelArgs:`
`29`	`29`	`default=None, metadata={"help": "The reranker name or path."}`
`30`	`30`	`)`
`31`	`31`	`reranker_model_class: Optional[str] = field(`
`32`		`- default="auto", metadata={"help": "The reranker model class. Available classes: ['auto', 'encoder-only-base', 'decoder-only-base', 'decoder-only-layerwise', 'decoder-only-lightweight']. Default: auto.", "choices": ["auto", "encoder-only-base", "decoder-only-base", "decoder-only-layerwise", "decoder-only-lightweight"]}`
	`32`	`+ default=None, metadata={"help": "The reranker model class. Available classes: ['auto', 'encoder-only-base', 'decoder-only-base', 'decoder-only-layerwise', 'decoder-only-lightweight']. Default: auto.", "choices": ["encoder-only-base", "decoder-only-base", "decoder-only-layerwise", "decoder-only-lightweight"]}`
`33`	`33`	`)`
`34`	`34`	`reranker_peft_path: Optional[str] = field(`
`35`	`35`	`default=None, metadata={"help": "The reranker peft path."}`