Skip to content

Commit 68f984e

Browse files
committed
update meta help info for add_reranker_score.py
1 parent df34cae commit 68f984e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

scripts/add_reranker_score.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ class ModelArgs:
2929
default=None, metadata={"help": "The reranker name or path."}
3030
)
3131
reranker_model_class: Optional[str] = field(
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"]}
32+
default=None, metadata={"help": "The reranker model class. Available classes: ['encoder-only-base', 'decoder-only-base', 'decoder-only-layerwise', 'decoder-only-lightweight']. Default: None. For the custom model, you need to specify the model class.", "choices": ["encoder-only-base", "decoder-only-base", "decoder-only-layerwise", "decoder-only-lightweight"]}
3333
)
3434
reranker_peft_path: Optional[str] = field(
3535
default=None, metadata={"help": "The reranker peft path."}

0 commit comments

Comments
 (0)