diff options
| author | Jonas Knobloch <jonas.knobloch@t-online.de> | 2026-03-17 21:00:38 +0100 |
|---|---|---|
| committer | Jonas Knobloch <jonas.knobloch@t-online.de> | 2026-03-17 21:01:25 +0100 |
| commit | af36446e68946c8b5ec0ac406c7544e204e42e46 (patch) | |
| tree | f49c5536679b56f143d19ca0b4bc7bb3aac6553e /bpc/cmd | |
| parent | d5f8cb4c28f636cc05386babcb0555b00dda83f7 (diff) | |
Remove bpc module
Diffstat (limited to 'bpc/cmd')
| -rw-r--r-- | bpc/cmd/bpc/main.go | 48 |
1 files changed, 0 insertions, 48 deletions
diff --git a/bpc/cmd/bpc/main.go b/bpc/cmd/bpc/main.go deleted file mode 100644 index 567b68a..0000000 --- a/bpc/cmd/bpc/main.go +++ /dev/null @@ -1,48 +0,0 @@ -package main - -import ( - "log" - "os" - - "github.com/jonasknobloch/mbpe" - - "go.jknobloc.com/x/bpc" - "go.jknobloc.com/x/gpt2" -) - -func main() { - m := model() - - if err := m.Init(); err != nil { - log.Fatal(err) - } - - t := tokenizer() - - bpc.Run(m, t) - - if err := m.Destroy(); err != nil { - log.Fatal(err) - } -} - -func model() *gpt2.Model { - return gpt2.NewModel("../gpt2/models/base/model.onnx", os.Getenv("BPC_CUDA_DEVICE_ID")) -} - -func tokenizer() *mbpe.Tokenizer { - m := mbpe.NewMBPE() - - if err := m.Load("../gpt2/models/base/vocab.json", "../gpt2/models/base/merges.txt"); err != nil { - panic(err) - } - - t := mbpe.NewTokenizer(m) - - byteLevel := mbpe.NewByteLevel(false) - - t.SetPreTokenizer(byteLevel) - t.SetDecoder(byteLevel) - - return t -} |
