disable the new NaN check for the CI

1 files changed, 2 insertions, 0 deletions

diff --git a/launch.py b/launch.py
index 715427fd..5afb2956 100644
--- a/launch.py
+++ b/launch.py
@@ -286,6 +286,8 @@ def tests(test_dir):
         sys.argv.append("./test/test_files/empty.pt")

     if "--skip-torch-cuda-test" not in sys.argv:

         sys.argv.append("--skip-torch-cuda-test")

+    if "--disable-nan-check" not in sys.argv:

+        sys.argv.append("--disable-nan-check")

 

     print(f"Launching Web UI in another process for testing with arguments: {' '.join(sys.argv[1:])}")