From c361b89026442f3412162657f330d500b803e052 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Tue, 17 Jan 2023 11:04:56 +0300
Subject: disable the new NaN check for the CI

---
 modules/devices.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'modules/devices.py')

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
 
-- 
cgit v1.2.1