Update app.py

app.py

@@ -156,17 +156,17 @@ def infer(
     print('-- filtered prompt --')
     print(enhanced_prompt)
     if latent_file:  # Check if a latent file is provided
-        initial_latents = pipe.prepare_latents(
-            batch_size=1,
-            num_channels_latents=pipe.transformer.in_channels,
-            height=pipe.transformer.config.sample_size[0],
-            width=pipe.transformer.config.sample_size[1],
-            dtype=pipe.transformer.dtype,
-            device=pipe.device,
-            generator=generator,
-        )
+      #  initial_latents = pipe.prepare_latents(
+      #      batch_size=1,
+      #      num_channels_latents=pipe.transformer.in_channels,
+      #      height=pipe.transformer.config.sample_size[0],
+       #     width=pipe.transformer.config.sample_size[1],
+      #      dtype=pipe.transformer.dtype,
+      #      device=pipe.device,
+      #      generator=generator,
+      #  )
         sd_image_a = torch.load(latent_file.name)  # Load the latent
-        initial_latents += sd_image_a
+      #  initial_latents += sd_image_a
         #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 --')
@@ -178,7 +178,7 @@ def infer(
             num_inference_steps=num_inference_steps,
             width=width,
             height=height,
-            latents=initial_latents,
+            latents=sd_image_a,
             generator=generator
           ).images[0]
     else: