llama : add RobertaForSequenceClassification reranker support (#13875)
This commit is contained in:
parent
1b8fb8152d
commit
6385b843a8
6 changed files with 24 additions and 8 deletions
|
@ -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,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue