Need attn_post_norm split for Muse-Flimmer

This commit is contained in:
Kawrakow
2026-08-11 10:37:22 +00:00
parent f091a1ad27
commit b39b20f0b8
+1 -1
View File
@@ -5259,7 +5259,7 @@ bool create_tensors_helper::create_tensors() {
if (layer.attn_norm) {
prepare_split_tensors(-1, ctx_split, layer.attn_norm, layer.split_attn_norm, mirror, mem_used);
}
if (is_gemma4_model() && layer.attn_post_norm) {
if ((is_gemma4_model() || model.arch == LLM_ARCH_MUSE_GLIMMER) && layer.attn_post_norm) {
prepare_split_tensors(-1, ctx_split, layer.attn_post_norm, layer.split_attn_post_norm, mirror, mem_used);
}
if (layer.rope_freqs) {