summaryrefslogtreecommitdiff
path: root/llm/cmd/eval/main.go
diff options
context:
space:
mode:
authorJonas Knobloch <jonas.knobloch@t-online.de>2026-05-04 15:19:01 +0200
committerJonas Knobloch <jonas.knobloch@t-online.de>2026-05-04 15:28:03 +0200
commit3150cbb7cf7c6febbf24ce73a9cb2ab048670607 (patch)
tree0b5e0a832abc32a1c3dc3e1525b812d5ea89c462 /llm/cmd/eval/main.go
parent0044303c62b876c5a3b1ca9c29ce079d30bce480 (diff)
Rename default config
Diffstat (limited to 'llm/cmd/eval/main.go')
-rw-r--r--llm/cmd/eval/main.go2
1 files changed, 1 insertions, 1 deletions
diff --git a/llm/cmd/eval/main.go b/llm/cmd/eval/main.go
index bd734b3..3a710ac 100644
--- a/llm/cmd/eval/main.go
+++ b/llm/cmd/eval/main.go
@@ -40,7 +40,7 @@ func model() *gpt2.Model {
WithLogProbs: true,
}
- m := gpt2.NewModel(shelf.Abs("models/gpt2/model_eval.onnx"), gpt2.DefaultConfig(), opts)
+ m := gpt2.NewModel(shelf.Abs("models/gpt2/model_eval.onnx"), gpt2.ConfigDefault(), opts)
if err := m.Init(); err != nil {
log.Fatal(err)