fix: GLM-DSA crash in llama-tokenize when using vocab_only (#22102)
* llama: fix crash in print_info for GLM-DSA when vocab_only is set * addressed code review comments * cont : simplify --------- Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
This commit is contained in:
co-authored by
Georgi Gerganov
parent
de71b5f81c
commit
81df3f7cfa
+2
-2
@@ -8180,11 +8180,10 @@ void llama_model::print_info() const {
|
|||||||
LLAMA_LOG_INFO("%s: n_cls_out = %u\n", __func__, hparams.n_cls_out);
|
LLAMA_LOG_INFO("%s: n_cls_out = %u\n", __func__, hparams.n_cls_out);
|
||||||
|
|
||||||
size_t i = 0;
|
size_t i = 0;
|
||||||
for (auto label : classifier_labels) {
|
for (const auto & label : classifier_labels) {
|
||||||
LLAMA_LOG_INFO("%s: cls_label[%2zu] = %s\n", __func__, i++, label.c_str());
|
LLAMA_LOG_INFO("%s: cls_label[%2zu] = %s\n", __func__, i++, label.c_str());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
if (arch == LLM_ARCH_MAMBA ||
|
if (arch == LLM_ARCH_MAMBA ||
|
||||||
arch == LLM_ARCH_MAMBA2 ||
|
arch == LLM_ARCH_MAMBA2 ||
|
||||||
@@ -8289,6 +8288,7 @@ void llama_model::print_info() const {
|
|||||||
LLAMA_LOG_INFO("%s: n_group_experts = %d\n", __func__, hparams.n_group_experts);
|
LLAMA_LOG_INFO("%s: n_group_experts = %d\n", __func__, hparams.n_group_experts);
|
||||||
LLAMA_LOG_INFO("%s: expert_group_scale = %.2f\n", __func__, hparams.expert_group_scale);
|
LLAMA_LOG_INFO("%s: expert_group_scale = %.2f\n", __func__, hparams.expert_group_scale);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
vocab.print_info();
|
vocab.print_info();
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user