From a006507a24b93a6af7e3cff8833fb96b48b159d7 Mon Sep 17 00:00:00 2001
From: Jonas Knobloch <jonas.knobloch@t-online.de>
Date: Wed, 8 Apr 2026 22:14:43 +0200
Subject: Refactor model config

---
 gpt2/config.go | 26 ++++++++++----------------
 1 file changed, 10 insertions(+), 16 deletions(-)

(limited to 'gpt2/config.go')

diff --git a/gpt2/config.go b/gpt2/config.go
index be7db30..b77578d 100644
--- a/gpt2/config.go
+++ b/gpt2/config.go
@@ -1,25 +1,19 @@
 package gpt2
 
 type Config struct {
-	vocabSize  int
-	nLayers    int
-	nHeads     int
-	headDim    int
-	nPositions int
+	VocabSize    int
+	NumLayers    int
+	NumHeads     int
+	HeadDim      int
+	NumPositions int
 }
 
 func DefaultConfig() Config {
 	return Config{
-		vocabSize:  50257,
-		nLayers:    12,
-		nHeads:     12,
-		headDim:    64,
-		nPositions: 1024,
+		VocabSize:    50257,
+		NumLayers:    12,
+		NumHeads:     12,
+		HeadDim:      64,
+		NumPositions: 1024,
 	}
 }
-
-func (c Config) WithVocabSize(vocabSize int) Config {
-	c.vocabSize = vocabSize
-
-	return c
-}
-- 
cgit v1.3.1