From d220be31e1e3082e56f9c59a735895839c99abce Mon Sep 17 00:00:00 2001
From: Void Agent <void@jayrup.hermes>
Date: Sun, 2 Aug 2026 14:25:14 +0100
Subject: test_jlens_v3: move tiny model to DEV (fix cpu/cuda mismatch on cuda
 runs)

---
 tests/test_jlens_v3.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_jlens_v3.py b/tests/test_jlens_v3.py
index 5ac3f67..4ebeaf3 100644
--- a/tests/test_jlens_v3.py
+++ b/tests/test_jlens_v3.py
@@ -31,7 +31,7 @@ def _tiny_model(n_layer=2, d=16, block=32):
     torch.manual_seed(0)
     cfg = GPTConfig(n_layer=n_layer, n_head=2, n_embd=d, block_size=block,
                     bias=False, vocab_size=65, dropout=0.0)
-    return GPT(cfg).eval()
+    return GPT(cfg).eval().to(DEV)
 
 
 def _tiny_data(n=2000, block=32):
-- 
cgit v1.2.3