From d703fa9fa5c10a34789f6ea6be3cbb9a5ba4249d Mon Sep 17 00:00:00 2001 From: Francis Couture-Harpin Date: Sat, 25 May 2024 12:47:01 -0400 Subject: [PATCH] convert-hf : fix flake8 indentation lint --- convert-hf-to-gguf.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 611d734c9..0dff1fad8 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -869,11 +869,11 @@ class BaichuanModel(Model): if bid is not None and name == f"model.layers.{bid}.self_attn.W_pack.weight": logger.info(f"Unpacking and permuting layer {bid}") yield (self.format_tensor_name(gguf.MODEL_TENSOR.ATTN_Q, bid), - self._reverse_hf_permute_part(data_torch, 0, head_count, head_count)) + self._reverse_hf_permute_part(data_torch, 0, head_count, head_count)) yield (self.format_tensor_name(gguf.MODEL_TENSOR.ATTN_K, bid), - self._reverse_hf_permute_part(data_torch, 1, head_count, head_count_kv)) + self._reverse_hf_permute_part(data_torch, 1, head_count, head_count_kv)) yield (self.format_tensor_name(gguf.MODEL_TENSOR.ATTN_V, bid), - self._reverse_hf_part(data_torch, 2)) + self._reverse_hf_part(data_torch, 2)) else: yield (self.map_tensor_name(name), data_torch)