Update app.py

app.py

@@ -156,7 +156,8 @@ def infer(
     print('-- filtered prompt --')
     print(enhanced_prompt)
     if latent_file:  # Check if a latent file is provided
-        sd_image_a = torch.load(latent_file.name)  # Load the latent
+        sd_image_a = torch.load(latent_file.name).to('cuda')  # Load the latent
+        sd_image_b = 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():
@@ -167,7 +168,7 @@ def infer(
             num_inference_steps=num_inference_steps,
             width=width,
             height=height,
-            latents=sd_image_a,
+            latents=sd_image_b,
             generator=generator
           ).images[0]
     else: