llama : support multiple classifier outputs and labels (#13940)
This commit is contained in:
1 parent
1caae7fc6c
commit
d17a809ef0
6 files changed
+101
-24
No files matched your search
@@ -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;
|
||||
|
||||
Reference in new issue
Block a user