update

app.py

@@ -88,12 +88,16 @@ class CompVisDenoiser(K.external.CompVisDenoiser):
         print("c_out.device:", c_out.device)
         print("sigma.device:", sigma.device)
         # eps_0, eps_1 = self.get_eps(input_0 * c_in, input_1 * c_in, self.sigma_to_t(sigma), **kwargs)
-        eps_0, eps_1 = self.get_eps(input_0 * c_in, self.sigma_to_t(sigma.cpu()).cuda(), **kwargs)
+        eps_0, eps_1 = self.get_eps(input_0 * c_in, self.sigma_to_t(sigma.cpu().float()).cuda(), **kwargs)
         
         return input_0 + eps_0 * c_out, eps_1
 
 def to_d(x, sigma, denoised):
     """Converts a denoiser output to a Karras ODE derivative."""
+
+    print(x.device)
+    print(denoised.device)
+    print(sigma.device)
     return (x - denoised) / append_dims(sigma, x.ndim)
 
 def default_noise_sampler(x):
@@ -206,7 +210,7 @@ def generate(
         print("uncond['c_crossattn'][0].device:", uncond["c_crossattn"][0].device)
         print("uncond['c_concat'][0].device:", uncond["c_concat"][0].device)
 
-        sigmas = model_wrap.get_sigmas(steps)
+        sigmas = model_wrap.get_sigmas(steps).to(model.device)
 
         extra_args = {
             "cond": cond,