From f4a9f9c262f7f8dbe52e8377a37716abc4f6c444 Mon Sep 17 00:00:00 2001
From: Haoming <hmstudy02@gmail.com>
Date: Tue, 26 May 2026 12:57:21 +0800
Subject: [PATCH] :(

---
 modules/devices.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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):