From cc95fe408bae1145360ed49245442137ddbe8b0f Mon Sep 17 00:00:00 2001
From: rossiyareich <tawinjira@hotmail.com>
Date: Sun, 18 Aug 2024 03:28:05 +0700
Subject: [PATCH] Fix scipy samplers on MPS

---
 nodes/nodes_rk_sampler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nodes/nodes_rk_sampler.py b/nodes/nodes_rk_sampler.py
index 1164b6a..cab5fa5 100644
--- a/nodes/nodes_rk_sampler.py
+++ b/nodes/nodes_rk_sampler.py
@@ -268,7 +268,7 @@ class RungeKuttaSamplerImpl:
         return samples
 
     def _call_scipy(self, model, x: torch.Tensor, sigmas: torch.Tensor, extra_args=None, callback=None, disable=None):
-        c_device = "mps" if HAS_MPS else "cpu"
+        c_device = "cpu"
         c_dtype = np.float32 if (HAS_MPS or HAS_DML) else np.float64
         o_device = x.device
         o_dtype = x.dtype