summaryrefslogtreecommitdiff
path: root/bpc/cmd
diff options
context:
space:
mode:
authorJonas Knobloch <jonas.knobloch@t-online.de>2026-03-17 21:00:38 +0100
committerJonas Knobloch <jonas.knobloch@t-online.de>2026-03-17 21:01:25 +0100
commitaf36446e68946c8b5ec0ac406c7544e204e42e46 (patch)
treef49c5536679b56f143d19ca0b4bc7bb3aac6553e /bpc/cmd
parentd5f8cb4c28f636cc05386babcb0555b00dda83f7 (diff)
Remove bpc module
Diffstat (limited to 'bpc/cmd')
-rw-r--r--bpc/cmd/bpc/main.go48
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
-}