diff options
| author | Jonas Knobloch <jonas.knobloch@t-online.de> | 2026-04-06 22:59:50 +0200 |
|---|---|---|
| committer | Jonas Knobloch <jonas.knobloch@t-online.de> | 2026-04-06 23:28:04 +0200 |
| commit | b9faf92148c7e6432ff0411783f444b7779fa36a (patch) | |
| tree | 96904cd4cf053b3d272b5f2b73c1141483e33863 /llm/cmd/eval | |
| parent | 58bdf1ac7155f4c5076cf4a077a5f39dfdedbac7 (diff) | |
Handle environment initialization on package level
Diffstat (limited to 'llm/cmd/eval')
| -rw-r--r-- | llm/cmd/eval/main.go | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/llm/cmd/eval/main.go b/llm/cmd/eval/main.go index 78a865e..43752d9 100644 --- a/llm/cmd/eval/main.go +++ b/llm/cmd/eval/main.go @@ -9,8 +9,16 @@ import ( ) func main() { + if err := gpt2.InitializeEnvironment(); err != nil { + log.Fatal(err) + } + perplexity() // logprobs() + + if err := gpt2.DestroyEnvironment(); err != nil { + log.Fatal(err) + } } func data() *dataset.ParquetReader { |