From a0b7f0b61c252dab735ca0c71accf17ec4cff877 Mon Sep 17 00:00:00 2001
From: Jodh Singh <jetjodh@gmail.com>
Date: Thu, 29 Aug 2024 10:50:38 -0400
Subject: [PATCH] fix for wrong device error

---
 sampling.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/sampling.py b/sampling.py
index 9970804..da5a1ce 100644
--- a/sampling.py
+++ b/sampling.py
@@ -89,7 +89,8 @@ def execute(self, model, latent_image, main_seed, steps, cfg, sampler_name, sche
         sigmas = sampler.sigmas
         sigma = sigmas[start_at_step] - sigmas[end_at_step]
         sigma /= model.model.latent_format.scale_factor
-        sigma = sigma.detach().cpu().item()
+        # Ensure sigma is on the same device as other tensors
+        sigma = torch.tensor(sigma, device=slerp_noise.device)
 
         work_latent = latent_image.copy()
         work_latent["samples"] = latent_image["samples"].clone() + slerp_noise * sigma
@@ -604,4 +605,4 @@ def execute(self, images, params, order_by, cols_value, cols_num, add_prompt, ad
     "TextEncodeForSamplerParams+": "🔧Text Encode for Sampler Params",
     "SamplerSelectHelper+": "🔧 Sampler Select Helper",
     "SchedulerSelectHelper+": "🔧 Scheduler Select Helper",
-}
\ No newline at end of file
+}