From e579fab4d06231785d10880068a4498e1dc9e025 Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Tue, 6 Feb 2024 17:46:23 -0800
Subject: [PATCH] try solve #71

---
 modules/sd_samplers_cfg_denoiser.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/modules/sd_samplers_cfg_denoiser.py b/modules/sd_samplers_cfg_denoiser.py
index 0a243630..a1ae8c3b 100644
--- a/modules/sd_samplers_cfg_denoiser.py
+++ b/modules/sd_samplers_cfg_denoiser.py
@@ -154,6 +154,9 @@ class CFGDenoiser(torch.nn.Module):
         if state.interrupted or state.skipped:
             raise sd_samplers_common.InterruptedException
 
+        original_x_device = x.device
+        original_x_dtype = x.dtype
+
         if self.classic_ddim_eps_estimation:
             acd = self.inner_model.inner_model.alphas_cumprod
             fake_sigmas = ((1 - acd) / acd) ** 0.5
@@ -195,5 +198,5 @@ class CFGDenoiser(torch.nn.Module):
             eps = (x - denoised) / sigma
             return eps
 
-        return denoised
+        return denoised.to(device=original_x_device, dtype=original_x_dtype)