Skip to content

Commit ffdd051

Browse files
committed
convert : update GGML script to use VocabFactory
Fixes #4631
1 parent cb4605f commit ffdd051

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

convert-llama-ggml-to-gguf.py

+4-8
Original file line numberDiff line numberDiff line change
@@ -371,15 +371,11 @@ def handle_metadata(cfg, hp):
371371
params = convert.Params.loadOriginalParamsJson(fakemodel, orig_config_path)
372372
else:
373373
raise ValueError('Unable to load metadata')
374-
vocab = convert.load_vocab(
375-
cfg.vocab_dir if cfg.vocab_dir is not None else cfg.model_metadata_dir,
376-
cfg.vocabtype)
377-
# FIXME: Respect cfg.vocab_dir?
378-
svocab = gguf.SpecialVocab(cfg.model_metadata_dir,
379-
load_merges = cfg.vocabtype == 'bpe',
380-
n_vocab = vocab.vocab_size)
374+
vocab_path = Path(cfg.vocab_dir if cfg.vocab_dir is not None else cfg.model_metadata_dir)
375+
vocab_factory = convert.VocabFactory(vocab_path)
376+
vocab, special_vocab = vocab_factory.load_vocab(cfg.vocabtype, cfg.model_metadata_dir)
381377
convert.check_vocab_size(params, vocab)
382-
return (params, vocab, svocab)
378+
return params, vocab, special_vocab
383379

384380

385381
def handle_args():

0 commit comments

Comments
 (0)