update

app.py

@@ -19,6 +19,13 @@ from funcs import (
     get_latent_z,
     save_videos
 )
+if torch.cuda.is_available():
+  device = "cuda"
+elif torch.backends.mps.is_available():
+  device = "mps"
+else:
+  device = "cpu"
+
 
 def download_model():
     REPO_ID = 'Doubiiu/DynamiCrafter_1024'
@@ -43,7 +50,7 @@ def infer(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123):
     assert os.path.exists(ckpt_path), "Error: checkpoint Not Found!"
     model = load_model_checkpoint(model, ckpt_path)
     model.eval()
-    model = model.cuda()
+    model = model.to(device)
     save_fps = 8
 
     seed_everything(seed)
@@ -51,7 +58,10 @@ def infer(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123):
         transforms.Resize(min(resolution)),
         transforms.CenterCrop(resolution),
         ])
-    torch.cuda.empty_cache()
+    if device == "cuda":
+        torch.cuda.empty_cache()
+    elif device == "mps":
+        torch.mps.empty_cache()
     print('start:', prompt, time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time())))
     start = time.time()
     if steps > 60:
@@ -154,4 +164,4 @@ with gr.Blocks(analytics_enabled=False, css=css) as dynamicrafter_iface:
                         fn = infer
         )
 
-dynamicrafter_iface.queue(max_size=12).launch(show_api=True)
+dynamicrafter_iface.queue(max_size=12).launch(show_api=True)