diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-17 11:04:56 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-17 11:05:01 +0300 |
| commit | c361b89026442f3412162657f330d500b803e052 (patch) | |
| tree | 4bba2484cbb13c2f631a0207f75a989d9dc8a961 /modules/devices.py | |
| parent | 93d3b820d0c8b447d53553e15a149973e9890155 (diff) | |
disable the new NaN check for the CI
Diffstat (limited to 'modules/devices.py')
| -rw-r--r-- | modules/devices.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/devices.py b/modules/devices.py index 6f034948..206184fb 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -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 |