From bfa6f820e6edf70de3782033cae15957bed59801 Mon Sep 17 00:00:00 2001 From: Haoming Date: Sat, 30 May 2026 13:11:41 +0800 Subject: [PATCH] oops --- modules/devices.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/devices.py b/modules/devices.py index 60b8521d..98f479bf 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -61,6 +61,6 @@ class NansException(Exception): def test_for_nans(x: torch.Tensor, *args, **kwargs): if torch.isnan(x).any(): - memory_management.logger.warning("Encountered NaN in Latent" + "; Try --disable-sage" if memory_management.sage_enabled() else "") + memory_management.logger.warning("Encountered NaN in Latent" + ("; Try --disable-sage" if memory_management.sage_enabled() else "")) x.nan_to_num_(nan=0.0, posinf=1.0, neginf=0.0) # raise NansException