update

app.py

@@ -5,7 +5,13 @@ from SDXL.diff_pipe import StableDiffusionXLDiffImg2ImgPipeline
 from diffusers import DPMSolverMultistepScheduler
 
 NUM_INFERENCE_STEPS = 50
-device = "cuda"
+if torch.cuda.is_available():
+  device = "cuda"
+elif torch.backends.mps.is_available():
+  device = "mps"
+else:
+  device = "cpu"
+#device = "cuda"
 
 base = StableDiffusionXLDiffImg2ImgPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True

requirements.txt

@@ -80,8 +80,8 @@ sympy==1.12
 tokenizers==0.15.0
 tomlkit==0.12.0
 toolz==0.12.1
-torch==2.1.1
-torchvision==0.16.1
+#torch==2.1.1
+#torchvision==0.16.1
 tqdm==4.66.1
 transformers==4.35.2
 triton==2.1.0