py : try to fix flake stuff

ggerganov · Jan 13, 2024 · c3d64a0 · c3d64a0
1 parent fe25223
commit c3d64a0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
@@ -25,7 +25,6 @@
 
 # check for any of the given keys in the dictionary and return the value of the first key found
 def get_key_opts(d, keys):
-    vals = []
     for k in keys:
         if k in d:
             return d[k]
@@ -267,7 +266,6 @@ def _set_vocab_gpt2(self):
                 toktypes.append(gguf.TokenType.USER_DEFINED)
             elif reverse_vocab[i] in added_vocab:
                 tokens.append(reverse_vocab[i])
-                # check if tokenizer has added_tokens_decoder
                 if hasattr(tokenizer, "added_tokens_decoder"):
                     if tokenizer.added_tokens_decoder[i].special:
                         toktypes.append(gguf.TokenType.CONTROL)
@@ -1092,7 +1090,9 @@ def set_gguf_parameters(self):
         self.gguf_writer.add_head_count_kv(get_key_opts(self.hparams, ["n_head", "num_attention_heads"]))
         self.gguf_writer.add_layer_norm_eps(get_key_opts(self.hparams, ["layer_norm_epsilon", "layer_norm_eps"]))
         self.gguf_writer.add_rope_dimension_count(
-                int(get_key_opts(self.hparams, ["partial_rotary_factor"]) * get_key_opts(self.hparams, ["n_embd", "hidden_size"])) // get_key_opts(self.hparams, ["n_head", "num_attention_heads"]))
+                int(get_key_opts(self.hparams, ["partial_rotary_factor"]) *
+                    get_key_opts(self.hparams, ["n_embd", "hidden_size"])) //
+                get_key_opts(self.hparams, ["n_head", "num_attention_heads"]))
         self.gguf_writer.add_file_type(self.ftype)
         self.gguf_writer.add_add_bos_token(False)