Update app.py

app.py

@@ -157,7 +157,7 @@ def infer(
     print(enhanced_prompt)
     if latent_file:  # Check if a latent file is provided
         sd_image_a = torch.load(latent_file.name).to('cuda')  # Load the latent
-        sd_image_b = pipe.vae.encode(sd_image_a.to(torch.bfloat16)).latent_dist.sample().mul_(0.18215)
+        #sd_image_b = pipe.vae.encode(sd_image_a.to(torch.bfloat16)).latent_dist.sample().mul_(0.18215)
         print("-- using latent file --")
         print('-- generating image --')
         with torch.no_grad():
@@ -168,7 +168,7 @@ def infer(
             num_inference_steps=num_inference_steps,
             width=width,
             height=height,
-            latents=sd_image_b,
+            latents=sd_image_a,
             generator=generator
           ).images[0]
     else: