Update app.py

app.py

@@ -119,7 +119,7 @@ def load_and_prepare_model(model_id):
     )
     sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", algorithm_type="sde-dpmsolver++")
     pipe.scheduler=sched
-    pipe.to(device=device, dtype=torch.bfloat16)
+    pipe.to(device=device)
     return pipe
 
 # Preload and compile both models