Select between cuda or mps based on environment.

utils/pipes.py

@@ -5,20 +5,24 @@ from utils import tiling
 
 
 def create_stable_diffusion_xl_pipeline():
+    device = torch.device("cuda" if torch.cuda.is_available() else "mps")
+
     pipe = StableDiffusionXLPipeline.from_pretrained(
         "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
     )
-    pipe = pipe.to("mps")
+    pipe = pipe.to(device)
     pipe.vae.disable_tiling()
     tiling.enable_circular_tiling([pipe.vae, pipe.text_encoder, pipe.unet])
     return pipe
 
 
 def create_stable_diffusion_xl_img2img_pipe():
+    device = torch.device("cuda" if torch.cuda.is_available() else "mps")
+
     pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
         "stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
     )
-    pipe = pipe.to("mps")
+    pipe = pipe.to(device)
     pipe.vae.disable_tiling()
     tiling.enable_circular_tiling([pipe.vae, pipe.text_encoder_2, pipe.unet])
     return pipe