disable the new NaN check for the CI

This commit is contained in:
AUTOMATIC
2023-01-17 11:04:56 +03:00
parent 93d3b820d0
commit c361b89026
3 changed files with 6 additions and 0 deletions

View File

@@ -113,6 +113,9 @@ class NansException(Exception):
def test_for_nans(x, where):
from modules import shared
if shared.cmd_opts.disable_nan_check:
return
if not torch.all(torch.isnan(x)).item():
return