From 3d04d337caf4eef74b4b6d15b995161132811218 Mon Sep 17 00:00:00 2001 From: RunningLeon Date: Mon, 8 Jul 2024 10:32:57 +0800 Subject: [PATCH] fix lint --- convert_hf_to_gguf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py index 89685d6d9..d9d9be61b 100755 --- a/convert_hf_to_gguf.py +++ b/convert_hf_to_gguf.py @@ -2145,7 +2145,7 @@ class InternLM2Model(Model): toktype = SentencePieceTokenTypes.UNUSED elif tokenizer.IsByte(token_id): toktype = SentencePieceTokenTypes.BYTE - # take care of ununsed raw token + # take care of ununsed raw token if piece.startswith('[UNUSED'): toktype = SentencePieceTokenTypes.UNKNOWN @@ -2245,7 +2245,7 @@ class InternLM2Model(Model): if self.hparams["rope_scaling"].get("type") == "linear": self.gguf_writer.add_rope_scaling_type(gguf.RopeScalingType.LINEAR) self.gguf_writer.add_rope_scaling_factor(self.hparams["rope_scaling"]["factor"]) - + def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iterable[tuple[str, Tensor]]: num_heads = self.hparams["num_attention_heads"] num_kv_heads = self.hparams["num_key_value_heads"]