From 55731a1873bcd2ffa6a15379e9a40ee21bd6718f Mon Sep 17 00:00:00 2001
From: Tom Penfold <116175928+penfoldtj@users.noreply.github.com>
Date: Mon, 2 Mar 2026 11:22:27 +0000
Subject: [PATCH] Update gaussian.py

device issue
---
 xanesnet/utils/gaussian.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/xanesnet/utils/gaussian.py b/xanesnet/utils/gaussian.py
index 4d4546e3..0286c8ba 100644
--- a/xanesnet/utils/gaussian.py
+++ b/xanesnet/utils/gaussian.py
@@ -64,7 +64,7 @@ def __init__(
         self.register_buffer("centers", centers)
 
     def synthesize(self, coeffs: torch.Tensor):
-        return coeffs @ self.Phi.T
+        return coeffs @ self.Phi.to(coeffs.device).T
 
 
 def build_ridge_operator(Phi: Tensor, lam: float = 1e-2) -> Tensor: