diff options
Diffstat (limited to 'research/lesci')
| -rw-r--r-- | research/lesci/cmd/lesci/main.go | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/research/lesci/cmd/lesci/main.go b/research/lesci/cmd/lesci/main.go index 27c257a..8a99c18 100644 --- a/research/lesci/cmd/lesci/main.go +++ b/research/lesci/cmd/lesci/main.go @@ -69,7 +69,7 @@ func must[T any](v T, err error) T { } func model(name string, vocabSize int) (*gpt2.Model, error) { - cfg := gpt2.DefaultConfig() + cfg := gpt2.ConfigDefault() cfg.VocabSize = vocabSize + 1 |