Destroy session on model teardown

2 files changed, 8 insertions, 0 deletions

diff --git a/llm/cmd/eval/logprobs.go b/llm/cmd/eval/logprobs.go
index baac805..f39e351 100644
--- a/llm/cmd/eval/logprobs.go
+++ b/llm/cmd/eval/logprobs.go
@@ -65,6 +65,10 @@ func logprobs() {
 	}); err != nil {
 		log.Fatal(err)
 	}
+
+	if err := m.Destroy(); err != nil {
+		log.Fatal(err)
+	}
 }
 
 func prepare(name string) (*sql.Stmt, *sql.DB, error) {
diff --git a/llm/cmd/eval/ppl.go b/llm/cmd/eval/ppl.go
index 330e4ab..2394d83 100644
--- a/llm/cmd/eval/ppl.go
+++ b/llm/cmd/eval/ppl.go
@@ -57,6 +57,10 @@ func perplexity() {
 	ppl := math.Exp(avg)
 
 	fmt.Println(ppl)
+
+	if err := m.Destroy(); err != nil {
+		log.Fatal(err)
+	}
 }
 
 func joined() dataset.Reader {