diff --git a/comfy/ldm/flux/model.py b/comfy/ldm/flux/model.py index db2ef3d6e..846703d52 100644 --- a/comfy/ldm/flux/model.py +++ b/comfy/ldm/flux/model.py @@ -159,7 +159,6 @@ class Flux(nn.Module): if add is not None: img += add - if img.dtype == torch.float16: img = torch.nan_to_num(img, nan=0.0, posinf=65504, neginf=-65504)