Update app.py

app.py

@@ -6,7 +6,7 @@ from torch.utils.data import Dataset, DataLoader
 import gradio as gr
 import sys
 import uuid
-import tqdm
+import tqdm 
 sys.path.append(os.path.abspath(os.path.join("", "..")))
 import gc
 import warnings
@@ -191,7 +191,7 @@ def inference(net, prompt, negative_prompt, guidance_scale, ddim_steps, seed):
         latents = noise_scheduler.step(noise_pred, t, latents).prev_sample
             
     latents = 1 / 0.18215 * latents
-    image = vae.decode(latents.float()).sample
+    image = vae.decode(latents).sample
     image = (image / 2 + 0.5).clamp(0, 1)
     image = image.detach().cpu().float().permute(0, 2, 3, 1).numpy()[0]