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