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/config.go | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'gpt2/config.go')

diff --git a/gpt2/config.go b/gpt2/config.go
index b77578d..cd3984d 100644
--- a/gpt2/config.go
+++ b/gpt2/config.go
@@ -8,6 +8,12 @@ type Config struct {
 	NumPositions int
 }
 
+type Options struct {
+	WithCache    bool
+	WithLogits   bool
+	WithLogProbs bool
+}
+
 func DefaultConfig() Config {
 	return Config{
 		VocabSize:    50257,
-- 
cgit v1.3.1