Skip to content

Commit dcf51e0

Browse files
committed
feat: Log mamba params for Granite Hybrid
Branch: GraniteFour Signed-off-by: Gabe Goodhart <ghart@us.ibm.com>
1 parent 2b36420 commit dcf51e0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/llama-model.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5174,7 +5174,8 @@ void llama_model::print_info() const {
51745174
if (arch == LLM_ARCH_MAMBA ||
51755175
arch == LLM_ARCH_MAMBA2 ||
51765176
arch == LLM_ARCH_JAMBA ||
5177-
arch == LLM_ARCH_FALCON_H1) {
5177+
arch == LLM_ARCH_FALCON_H1 ||
5178+
arch == LLM_ARCH_GRANITE_HYBRID) {
51785179
LLAMA_LOG_INFO("%s: ssm_d_conv = %u\n", __func__, hparams.ssm_d_conv);
51795180
LLAMA_LOG_INFO("%s: ssm_d_inner = %u\n", __func__, hparams.ssm_d_inner);
51805181
LLAMA_LOG_INFO("%s: ssm_d_state = %u\n", __func__, hparams.ssm_d_state);

0 commit comments

Comments
 (0)