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(torch_device="cuda", torch_dtype=torch.bfloat16)
+    pipe.to(device)
     return pipe
 
 # Preload and compile both models
@@ -340,7 +340,7 @@ with gr.Blocks(css=css) as demo:
                 minimum=0.1,
                 maximum=6,
                 step=0.1,
-                value=3.0,
+                value=4.2,
             )
             num_inference_steps = gr.Slider(
                 label="Number of inference steps",