fix v granularity

This commit is contained in:
Aman Gupta
2026-09-24 15:28:59 +08:00
parent e52cdd8cd5
commit c645b16990
+1 -1
View File
@@ -768,7 +768,7 @@ struct ggml_backend_meta_split_state llama_meta_device_get_split_state(const str
}
if (std::regex_match(tensor_name, pattern_attn_out_weight)) {
GGML_ASSERT(segments.size() == 1);
return {granularity_q};
return {granularity_head * hparams.n_embd_head_v(il)};
}
if (std::regex_match(tensor_name, pattern_attn_gate_weight)) {
GGML_ASSERT(segments.size() == 1);