1 files changed, 2 insertions, 0 deletions
diff --git a/launch.py b/launch.py
index 8ed1dffc..ed21d0e6 100644
--- a/launch.py
+++ b/launch.py
@@ -244,6 +244,8 @@ def tests(test_dir):
     if "--ckpt" not in sys.argv:
         sys.argv.append("--ckpt")
         sys.argv.append("./test/test_files/empty.pt")
+    if "--skip-torch-cuda-test" not in sys.argv:
+        sys.argv.append("--skip-torch-cuda-test")
 
     print(f"Launching Web UI in another process for testing with arguments: {' '.join(sys.argv[1:])}")