Skip to content

Commit 56ff48c

Browse files
committed
fix lint error
Signed-off-by: XingXing Qiao <[email protected]>
1 parent 1f908b1 commit 56ff48c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

convert-hf-to-gguf.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2427,6 +2427,7 @@ def set_vocab(self, *args, **kwargs):
24272427
self.gguf_writer.add_add_bos_token(True)
24282428
self.gguf_writer.add_add_eos_token(True)
24292429

2430+
24302431
@Model.register("ChatGLMModel")
24312432
class ChatGLMModel(Model):
24322433
model_arch = gguf.MODEL_ARCH.CHATGLM
@@ -2443,8 +2444,6 @@ def set_vocab(self):
24432444
vocab_size = hparams.get("padded_vocab_size", len(tokenizer.get_vocab()))
24442445
assert max(tokenizer.get_vocab().values()) < vocab_size
24452446

2446-
reverse_vocab = {id_: encoded_tok for encoded_tok, id_ in tokenizer.get_vocab().items()}
2447-
24482447
for token_id in range(vocab_size):
24492448
piece = tokenizer._convert_id_to_token(token_id)
24502449
if token_id == 0:

0 commit comments

Comments
 (0)