diff options
Diffstat (limited to 'llm/cmd/ppl')
| -rw-r--r-- | llm/cmd/ppl/ppl.go | 70 |
1 files changed, 0 insertions, 70 deletions
diff --git a/llm/cmd/ppl/ppl.go b/llm/cmd/ppl/ppl.go deleted file mode 100644 index 6912e48..0000000 --- a/llm/cmd/ppl/ppl.go +++ /dev/null @@ -1,70 +0,0 @@ -package main - -import ( - "fmt" - "log" - - "github.com/jonasknobloch/mbpe" - - "go.jknobloc.com/x/dataset" - "go.jknobloc.com/x/gpt2" - "go.jknobloc.com/x/llm" -) - -func main() { - d := data() - m := model() - t := tokenizer() - - e := llm.NewEvaluator() - - e.SetTokenizer(t) - e.AddModel(m) - - ppl, err := e.Perplexity(d, 1024, 512) - - if err != nil { - log.Fatal(err) - } - - fmt.Println(ppl) -} - -func data() *dataset.ParquetReader { - var miniPile *dataset.ParquetReader - - if r, err := dataset.NewParquetReader("dataset/cmd/dataset/tmp/minipile/validation"); err != nil { - log.Fatal(err) - } else { - miniPile = r - } - - return miniPile -} - -func model() *gpt2.Model { - m := gpt2.NewModel("gpt2/models/base/model.onnx", "0") - - if err := m.Init(); err != nil { - log.Fatal(err) - } - - return m -} - -func tokenizer() *mbpe.Tokenizer { - m := mbpe.NewMBPE() - - if err := m.Load("gpt2/models/base/vocab.json", "gpt2/models/base/merges.txt"); err != nil { - log.Fatal(err) - } - - t := mbpe.NewTokenizer(m) - - byteLevel := mbpe.NewByteLevel(false) - - t.SetPreTokenizer(byteLevel) - t.SetDecoder(byteLevel) - - return t -} |
