diff --git a/modules/devices.py b/modules/devices.py
index df003a58..60b8521d 100644
--- a/modules/devices.py
+++ b/modules/devices.py
@@ -55,8 +55,8 @@ def without_autocast(*args, **kwargs):
     return contextlib.nullcontext()
 
 
-# class NansException(Exception):
-#     pass
+class NansException(Exception):
+    pass
 
 
 def test_for_nans(x: torch.Tensor, *args, **kwargs):