298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-10 09:27:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix black images on some flux models in fp16. (#8495)

Browse Source
This commit is contained in:
comfyanonymous 2025-06-11 12:09:11 -07:00 committed by GitHub
parent 373a9386a4
commit af1eb58be8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/ldm/flux/model.py
View File

@ -187,6 +187,9 @@ class Flux(nn.Module):
if add is not None:
img[:, txt.shape[1] :, ...] += add
if img.dtype == torch.float16:
img = torch.nan_to_num(img, nan=0.0, posinf=65504, neginf=-65504)
img = img[:, txt.shape[1] :, ...]
img = self.final_layer(img, vec) # (N, T, patch_size ** 2 * out_channels)