Update fn.py

fn.py

@@ -38,7 +38,9 @@ def load_model(_model = None, _vae = None, loras = []):
     for lora in loras:
         pipe.load_lora_weights(".", weight_name=lora + ".safetensors")
 
-    pipe.to("cuda")
+    if torch.cuda.is_available():
+        pipe.to("cuda")
+
     pipe.enable_vae_slicing()
 
 def pil_to_webp(img):