Update app.py

app.py

@@ -22,10 +22,10 @@ def generate(prompt, negative_prompt, steps):
         guidance_scale=4.0,
         num_images_per_prompt=1,
         num_inference_steps=steps
-    ).to(dtype=torch.bfloat16)
+    )
     
     decoder_output = decoder(
-        image_embeddings=prior_output.image_embeddings.to(torch.float16),
+        image_embeddings=prior_output.image_embeddings.to(torch.bfloat16),
         prompt=prompt,
         guidance_scale=0.0,
         output_type="pil",