Merge branch 'comfyanonymous:master' into master

This commit is contained in:
patientx 2024-10-20 10:20:06 +03:00 committed by GitHub
commit 1678ea8f9c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -41,6 +41,8 @@ def manual_stochastic_round_to_float8(x, dtype, generator=None):
(2.0 ** (-EXPONENT_BIAS + 1)) * abs_x
)
inf = torch.finfo(dtype)
torch.clamp(sign, min=inf.min, max=inf.max, out=sign)
return sign