update

app.py

@@ -23,8 +23,16 @@ snapshot_download(
 
 # global variable
 MAX_SEED = np.iinfo(np.int32).max
-device = "cuda" if torch.cuda.is_available() else "cpu"
-dtype = torch.float16 if str(device).__contains__("cuda") else torch.float32
+#device = "cuda" if torch.cuda.is_available() else "cpu"
+#dtype = torch.float16 if str(device).__contains__("cuda") else torch.float32
+if torch.cuda.is_available():
+    device = torch.device("cuda")
+elif torch.backends.mps.is_available():
+    device = torch.device("mps")
+else:
+    device = torch.device("cpu")
+dtype = torch.float16 if str(device).__contains__("cuda") or str(device).__contains__("mps") else torch.float32
+
 
 # initialization
 base_model_path = "stabilityai/stable-diffusion-xl-base-1.0"

requirements.txt

@@ -1,6 +1,6 @@
 diffusers==0.27.2
-torch>=2.0.0
-torchvision>=0.15.1
+#torch>=2.0.0
+#torchvision>=0.15.1
 transformers>=4.37.1
 accelerate
 safetensors
@@ -13,4 +13,4 @@ opencv-python
 gradio
 controlnet_aux
 gdown
-peft
+peft