Update app.py

app.py

@@ -130,8 +130,7 @@ def generate(segment, image, prompt, size, guidance_scale, num_inference_steps,
             dtype=torch.float32, device=device, generator=generator, latents=None,
         )
         image_latents = pipe.image_latents(
-                #image, 1, size, size, device, torch.float32, num_channels_latents, num_latent_frames
-                image, 1, size, size, device, torch.bfloat16, num_channels_latents, num_latent_frames
+                image, 1, size, size, device, torch.float32, num_channels_latents, num_latent_frames
         )
         image_latents = image_latents.to("cuda", pipe.transformer.dtype)
         pipe.vae.to("cpu")