mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-07-21 21:01:24 +08:00
oops
This commit is contained in:
parent
a4b817ef82
commit
bfa6f820e6
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user