From b9faf92148c7e6432ff0411783f444b7779fa36a Mon Sep 17 00:00:00 2001
From: Jonas Knobloch <jonas.knobloch@t-online.de>
Date: Mon, 6 Apr 2026 22:59:50 +0200
Subject: Handle environment initialization on package level

---
 llm/cmd/eval/main.go | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'llm/cmd')

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 {
-- 
cgit v1.3.1