Merge pull request #135 from pytorch-labs/mikekgfb-mps-1

Update generate.py to enable MPS support

Author: mikekgfb

generate.py

@@ -16,7 +16,7 @@
 def device_sync(device):
     if "cuda" in device:
         torch.cuda.synchronize(device)
-    elif "cpu" in device:
+    elif ("cpu" in device) or ("mps" in device):
         pass
     else:
         print(f"device={device} is not yet suppported")