check if cuda is available

Author: tcapelle

inference.py

@@ -25,8 +25,8 @@
     img_size=64, # image size to use
     num_random_experiments = 2, # we will perform inference multiple times on the same inputs
     seed=42,
-    # device="cuda:0",
-    device="mps",
+    device="cuda" if torch.cuda.is_available() else "cpu",
+    # device="mps",
     sampler="ddim",
     future_frames=10,  # number of future frames
     bs=8, # how many samples