From 29c52bf0f0981d4ea8e12e86237365da2b225ef7 Mon Sep 17 00:00:00 2001 From: Jonas Knobloch Date: Mon, 4 May 2026 15:39:45 +0200 Subject: Add parameter to set hidden dimension --- research/sander/cmd/sander/main.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'research/sander/cmd') diff --git a/research/sander/cmd/sander/main.go b/research/sander/cmd/sander/main.go index 56f1488..ae7f7b9 100644 --- a/research/sander/cmd/sander/main.go +++ b/research/sander/cmd/sander/main.go @@ -5,6 +5,7 @@ import ( "log" "path" + "go.jknobloc.com/x/gpt2" "go.jknobloc.com/x/research/sander" "go.jknobloc.com/x/shelf" "go.jknobloc.com/x/tokenizer/bpe" @@ -33,7 +34,7 @@ func run(src, dst string) error { } t := must(bpe.NewTokenizerFromFiles(path.Join(src, "vocab.json"), path.Join(src, "merges.txt"), cfg)) - e := must(sander.NewExperiment(dst, path.Join(src, "model.onnx"), sander.UnusedTokensMBPE(t))) + e := must(sander.NewExperiment(dst, path.Join(src, "model.onnx"), sander.UnusedTokensMBPE(t), gpt2.ConfigDefault().HiddenDim)) return e.Run() } -- cgit v1.3.1