diff --git a/comfy/float.py b/comfy/float.py index 0c8f1d6c..521316fd 100644 --- a/comfy/float.py +++ b/comfy/float.py @@ -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