clarify the option to disable NaN check.

3 years ago · 9beb794e0b
parent 6f31d2210c
commit 9beb794e0b
1 changed files with 2 additions and 0 deletions
--- a/modules/devices.py
+++ b/modules/devices.py
@ -143,6 +143,8 @@ def test_for_nans(x, where):
    else:
        message = "A tensor with all NaNs was produced."

+    message += " Use --disable-nan-check commandline argument to disable this check."
+
    raise NansException(message)