From 4386784c1006321c5cf0e1e005b19e0a7d7563ae Mon Sep 17 00:00:00 2001
From: Jonas Knobloch <jonas.knobloch@t-online.de>
Date: Wed, 8 Apr 2026 22:24:10 +0200
Subject: Refactor model options

* Add options struct
---
 gpt2/model_test.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'gpt2/model_test.go')

diff --git a/gpt2/model_test.go b/gpt2/model_test.go
index a4de628..e562e0d 100644
--- a/gpt2/model_test.go
+++ b/gpt2/model_test.go
@@ -40,7 +40,13 @@ func fromModel() []float32 {
 }
 
 func model() *Model {
-	m := NewModel("models/base/model.onnx", "0", DefaultConfig(), true, true, false) // TODO check if CUDA is available
+	opts := Options{
+		WithCache:    true,
+		WithLogits:   true,
+		WithLogProbs: false,
+	}
+
+	m := NewModel("models/base/model.onnx", "0", DefaultConfig(), opts) // TODO check if CUDA is available
 
 	if err := m.Init(); err != nil {
 		log.Fatal(err)
-- 
cgit v1.3.1