Update app.py

app.py

@@ -230,7 +230,7 @@ class main():
             latents = self.noise_scheduler.step(noise_pred, t, latents).prev_sample
             
         latents = 1 / 0.18215 * latents
-        image = self.vae.decode(latents).sample
+        image = self.vae.decode(latents.float()).sample
         image = (image / 2 + 0.5).clamp(0, 1)
         image = image.detach().cpu().float().permute(0, 2, 3, 1).numpy()[0]