Update app.py

app.py

@@ -19,6 +19,18 @@ pipe.to(device)
 pipe.text_encoder.to(device, dtype=torch.float32)
 pipe.vae.to(device, dtype=torch.float32)
 pipe.unet.to(device, dtype=torch.float32)
+pipe.scheduler.to(device, dtype=torch.float32)
+
+def force_float32(model):
+    for param in model.parameters():
+        param.data = param.data.to(torch.float32)
+    for buffer in model.buffers():
+        buffer.data = buffer.data.to(torch.float32)
+
+force_float32(pipe.text_encoder)
+force_float32(pipe.vae)
+force_float32(pipe.unet)
+force_float32(pipe.scheduler)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
@@ -47,7 +59,7 @@ def infer(
     image = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
-        guidance_scale=guidance_scale,
+        guidance_scale=torch.tensor(guidance_scale, dtype=torch.float32, device=device),
         num_inference_steps=num_inference_steps,
         width=width,
         height=height,