llama : add RobertaForSequenceClassification reranker support (#13875)
This commit is contained in:
parent
1b8fb8152d
commit
6385b843a8
6 changed files with 24 additions and 8 deletions
|
@ -177,6 +177,9 @@ class Keys:
|
|||
EMBEDDING_LENGTH = "{arch}.convnext.embedding_length"
|
||||
BLOCK_COUNT = "{arch}.convnext.block_count"
|
||||
|
||||
class Classifier:
|
||||
OUTPUT_LABELS = "{arch}.classifier.output_labels"
|
||||
|
||||
class Tokenizer:
|
||||
MODEL = "tokenizer.ggml.model"
|
||||
PRE = "tokenizer.ggml.pre"
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue