Update app.py

app.py

@@ -330,7 +330,7 @@ def generate_images_60(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
         ).images
     
     # 2. Manually decode with our float32 VAE
-    latents_fp32 = latents.to(torch.float32)
+    latents_fp32 = sd_image.to(torch.float32)
     latents_fp32 = 1 / pipe.vae.config.scaling_factor * latents_fp32
     with torch.no_grad():
         # This is our high-precision sRGB tensor in range [-1, 1]
@@ -391,7 +391,7 @@ def generate_images_110(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
         ).images
     
     # 2. Manually decode with our float32 VAE
-    latents_fp32 = latents.to(torch.float32)
+    latents_fp32 = sd_image.to(torch.float32)
     latents_fp32 = 1 / pipe.vae.config.scaling_factor * latents_fp32
     with torch.no_grad():
         # This is our high-precision sRGB tensor in range [-1, 1]