Update app.py

app.py

@@ -203,8 +203,8 @@ def load_and_prepare_model():
     pipe.watermark=None
     pipe.safety_checker=None  
     #pipe.to(torch.device('cuda:0'), torch.bfloat16)
+    pipe.to(torch.bfloat16)
     pipe.to(accelerator.device)
-
     return pipe
 
 #hidet.option.parallel_build(False)
@@ -272,7 +272,8 @@ def generate_30(
     filename = uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
     upload_to_ftp(filename) 
     batch_options = options.copy()
-    rv_image = pipe(**batch_options).images[0]
+    with accelerator.autocast():
+        rv_image = pipe(**batch_options).images[0]
     sd_image_path = f"rv_C_{timestamp}.png"
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)