mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2025-12-09 22:14:34 +08:00
Small flux optimization. (#8611)
This commit is contained in:
parent
7e9267fa77
commit
f7fb193712
@ -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:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user