llama : support multiple classifier outputs and labels (#13940)

This commit is contained in:
Sigbjørn Skjæret authored and GitHub committed 2025-06-06 09:03:25 +02:00
1 parent 1caae7fc6c
commit d17a809ef0
6 files changed
+101 -24

No files matched your search

+3
View File
@@ -329,6 +329,9 @@ struct llama_model {
llama_hparams hparams = {};
llama_vocab vocab;
// for classifier models
std::vector<std::string> classifier_labels;
struct ggml_tensor * tok_embd = nullptr;
struct ggml_tensor * type_embd = nullptr;
struct ggml_tensor * pos_embd = nullptr;