convert : make lint happy
This commit is contained in:
parent
1eee98f01f
commit
8ff0991eed
1 changed files with 1 additions and 2 deletions
|
@ -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):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue