happyz
/
llama.cpp
mirror of https://github.com/ggerganov/llama.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

try to make lint happy

This commit is contained in:
Yee Man Chan 2026-01-11 22:33:58 +08:00
parent 58d1ee5227
commit 4f6ef2c085
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
convert_hf_to_gguf.py
View File

@ -5159,7 +5159,6 @@ class KimiLinearModel(TextModel):
self.gguf_writer.add_ssm_conv_kernel(ssm_d_conv) self.gguf_writer.add_ssm_conv_kernel(ssm_d_conv)
kda_head_dim = self.hparams.get("kda_head_dim") or linear_attn_config.get("head_dim") kda_head_dim = self.hparams.get("kda_head_dim") or linear_attn_config.get("head_dim")
if kda_head_dim is not None: if kda_head_dim is not None:
self.gguf_writer.add_kda_head_dim(kda_head_dim) self.gguf_writer.add_kda_head_dim(kda_head_dim)
@ -5328,11 +5327,7 @@ class KimiLinearModel(TextModel):
kv_b = data_torch.view(n_head_kv, v_head_dim + qk_nope_head_dim, data_torch.shape[-1]) kv_b = data_torch.view(n_head_kv, v_head_dim + qk_nope_head_dim, data_torch.shape[-1])
k_b, v_b = torch.split(kv_b, [qk_nope_head_dim, v_head_dim], dim=1) k_b, v_b = torch.split(kv_b, [qk_nope_head_dim, v_head_dim], dim=1)
k_b = k_b.transpose(1, 2) k_b = k_b.transpose(1, 2)
return [(self.map_tensor_name(name_kb), k_b), (self.map_tensor_name(name_vb), v_b)]
return [
(self.map_tensor_name(name_kb), k_b),
(self.map_tensor_name(name_vb), v_b)
]
mapped_name = self.map_tensor_name(name) mapped_name = self.map_tensor_name(name)
logger.info(f"Returning {mapped_name}: shape after = {tuple(data_torch.shape)}") logger.info(f"Returning {mapped_name}: shape after = {tuple(data_torch.shape)}")