jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Small flux optimization. (#8611)

Browse Source
This commit is contained in:
comfyanonymous
2025-06-20 02:37:32 -07:00
committed by GitHub
parent 7e9267fa77
commit f7fb193712
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/flux/layers.py
View File

@@ -118,7 +118,7 @@ class Modulation(nn.Module):
def apply_mod(tensor, m_mult, m_add=None, modulation_dims=None): def apply_mod(tensor, m_mult, m_add=None, modulation_dims=None):
if modulation_dims is None: if modulation_dims is None:
if m_add is not None: if m_add is not None:
return tensor * m_mult + m_add return torch.addcmul(m_add, tensor, m_mult)
else: else:
return tensor * m_mult return tensor * m_mult
else: else: