Update app.py

app.py

@@ -40,9 +40,11 @@ def generate(prompt, option, progress=gr.Progress()):
     print(prompt, option)
     ckpt, step = opts[option]
     progress((0, step))
-
-    pipe.to(device, dtype)
-    safety_checker.to(device, dtype)
+    
+    if pipe.device != device:
+        pipe.to(device, dtype)
+    if safety_checker.device != device:
+        safety_checker.to(device, dtype)
 
     if step != step_loaded:
         print(f"Switching checkpoint from {step_loaded} to {step}")