llama : add RobertaForSequenceClassification reranker support (#13875)

This commit is contained in:
Sigbjørn Skjæret
2025-05-29 08:15:01 +02:00
committed by GitHub
parent 1b8fb8152d
commit 6385b843a8
6 changed files with 24 additions and 8 deletions

View File

@ -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"