From a96c1bf68bf5c477e1d3f105c18346c80d80b47a Mon Sep 17 00:00:00 2001
From: Jonas Knobloch <jonas.knobloch@t-online.de>
Date: Sat, 4 Apr 2026 21:43:38 +0200
Subject: Add config struct

---
 gpt2/cmd/gpt2/main.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'gpt2/cmd')

diff --git a/gpt2/cmd/gpt2/main.go b/gpt2/cmd/gpt2/main.go
index a832ccd..cad3333 100644
--- a/gpt2/cmd/gpt2/main.go
+++ b/gpt2/cmd/gpt2/main.go
@@ -10,7 +10,7 @@ import (
 func main() {
 	prompt := []int64{464, 2068, 7586, 21831}
 
-	m := gpt2.NewModel("models/base/model.onnx", "")
+	m := gpt2.NewModel("gpt2/models/base/model.onnx", "0", gpt2.NewDefaultConfig())
 
 	if err := m.Init(); err != nil {
 		log.Fatal(err)
-- 
cgit v1.3.1