convert : make lint happy

This commit is contained in:
Stanisław Szymczyk 2025-01-28 11:02:52 +01:00
parent 1eee98f01f
commit 8ff0991eed

View file

@ -4148,7 +4148,7 @@ class DeepseekV2Model(Model):
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 = k_b.transpose(1, 2);
k_b = k_b.transpose(1, 2)
k_b = k_b.reshape(n_head_kv * data_torch.shape[-1], qk_nope_head_dim)
v_b = v_b.reshape(n_head_kv * v_head_dim, data_torch.shape[-1])
@ -4158,7 +4158,6 @@ class DeepseekV2Model(Model):
(self.map_tensor_name(name_vb), v_b)
]
return [(self.map_tensor_name(name), data_torch)]
def prepare_tensors(self):