Update utils.py

utils.py

@@ -230,7 +230,7 @@ def generate_image(prompt, method, seed, loss_apply=False):
     latents = torch.randn(
         (batch_size, pipe.unet.config.in_channels, height// 8, width //8),
         generator = generator,
-    ).to(torch.float16)
+    ).to(torch.float32)
 
 
     latents = latents.to(torch_device)
@@ -243,7 +243,7 @@ def generate_image(prompt, method, seed, loss_apply=False):
         latent_model_input = scheduler.scale_model_input(latent_model_input, t)
 
         with torch.no_grad():
-            noise_pred = pipe.unet(latent_model_input.to(torch.float16), t, encoder_hidden_states=text_embeddings)["sample"]
+            noise_pred = pipe.unet(latent_model_input.to(torch.float32), t, encoder_hidden_states=text_embeddings)["sample"]
 
         noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
         noise_pred = noise_pred_uncond + guidance_scale * (noise_pred_text - noise_pred_uncond)