Update model.py

This commit is contained in:
comfyanonymous 2025-06-11 15:13:04 -04:00 committed by GitHub
parent 12134fb9a3
commit 2b05e55340
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -159,7 +159,6 @@ class Flux(nn.Module):
if add is not None: if add is not None:
img += add img += add
if img.dtype == torch.float16: if img.dtype == torch.float16:
img = torch.nan_to_num(img, nan=0.0, posinf=65504, neginf=-65504) img = torch.nan_to_num(img, nan=0.0, posinf=65504, neginf=-65504)