diff options
Diffstat (limited to 'llmc/cmd/debug/debug.go')
| -rw-r--r-- | llmc/cmd/debug/debug.go | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/llmc/cmd/debug/debug.go b/llmc/cmd/debug/debug.go index d1a621a..f076949 100644 --- a/llmc/cmd/debug/debug.go +++ b/llmc/cmd/debug/debug.go @@ -16,7 +16,10 @@ func main() { _ = must(llmc.Deserialize(shelf.Abs("llmc/tinyshakespeare/tiny_shakespeare_train.bin"), &data)) _ = must(llmc.Deserialize(shelf.Abs("test/llmc/tinyshakespeare/tiny_shakespeare_train.bin"), &gold)) - t := must(bpe.NewTokenizerFromFiles(shelf.Abs("models/gpt2/vocab.json"), shelf.Abs("models/gpt2/merges.txt"))) + v := shelf.Abs("models/gpt2/vocab.json") + m := shelf.Abs("models/gpt2/merges.txt") + + t := must(bpe.NewTokenizerFromFiles(v, m, bpe.DefaultConfig())) itoa := bpe.Itoa(t) |