mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-04 01:37:10 +08:00
Update model.py
This commit is contained in:
parent
12134fb9a3
commit
2b05e55340
@ -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)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user