From 2b05e55340e147edad3894df759cf90e4bf52371 Mon Sep 17 00:00:00 2001 From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com> Date: Wed, 11 Jun 2025 15:13:04 -0400 Subject: [PATCH] Update model.py --- comfy/ldm/flux/model.py | 1 - 1 file changed, 1 deletion(-) 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)