From d35f5fad1d26fc09cfd98f6e7faf17d6a48e4e8f Mon Sep 17 00:00:00 2001
From: Jonas Knobloch <jonas.knobloch@t-online.de>
Date: Thu, 23 Apr 2026 20:14:26 +0200
Subject: Use platform specific defaults for CUDA device ID

---
 gpt2/model_test.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'gpt2/model_test.go')

diff --git a/gpt2/model_test.go b/gpt2/model_test.go
index e562e0d..051a41a 100644
--- a/gpt2/model_test.go
+++ b/gpt2/model_test.go
@@ -46,7 +46,7 @@ func model() *Model {
 		WithLogProbs: false,
 	}
 
-	m := NewModel("models/base/model.onnx", "0", DefaultConfig(), opts) // TODO check if CUDA is available
+	m := NewModel("models/base/model.onnx", DefaultConfig(), opts)
 
 	if err := m.Init(); err != nil {
 		log.Fatal(err)
-- 
cgit v1.3.1