From b2d8ba181025d4825dedbec7a88a5c8c6e8f98db Mon Sep 17 00:00:00 2001
From: Lars Bogner <larsbog@ml6.hpc.uio.no>
Date: Tue, 16 Dec 2025 18:52:14 +0100
Subject: [PATCH] edit predit

---
 Code/python/notebooks/predict.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Code/python/notebooks/predict.py b/Code/python/notebooks/predict.py
index df8f885..761e331 100644
--- a/Code/python/notebooks/predict.py
+++ b/Code/python/notebooks/predict.py
@@ -50,7 +50,7 @@ model = cls(
             device=DEVICE,
         )
 
-model.load_state_dict(torch.load("LSTM_wu900_ps150_aw0.001.pt"))
+model.load_state_dict(torch.load("LSTM_wu900_ps150_aw0.001.pt", map_location=torch.device('cpu')))
 model.to(DEVICE)
 model.eval()
 with torch.no_grad():