Skip to content

Commit

Permalink
convert : make lint happy
Browse files Browse the repository at this point in the history
  • Loading branch information
sszymczy committed Jan 28, 2025
1 parent 1eee98f commit 8ff0991
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions convert_hf_to_gguf.py
Original file line number Diff line number Diff line change
Expand Up @@ -4148,7 +4148,7 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter

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])

Expand All @@ -4158,7 +4158,6 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
(self.map_tensor_name(name_vb), v_b)
]


return [(self.map_tensor_name(name), data_torch)]

def prepare_tensors(self):
Expand Down

0 comments on commit 8ff0991

Please sign in to comment.