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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 24 additions and 8 deletions

View file

@ -213,6 +213,8 @@ enum llm_kv {
LLM_KV_CONVNEXT_EMBEDDING_LENGTH,
LLM_KV_CONVNEXT_BLOCK_COUNT,
LLM_KV_CLASSIFIER_OUTPUT_LABELS,
// deprecated:
LLM_KV_TOKENIZER_PREFIX_ID,
LLM_KV_TOKENIZER_SUFFIX_ID,