This commit is contained in:
Haoming 2026-05-30 13:11:41 +08:00
parent a4b817ef82
commit bfa6f820e6

View File

@ -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