Small flux optimization. (#8611)

This commit is contained in:
comfyanonymous 2025-06-20 02:37:32 -07:00 committed by GitHub
parent 81f38bce38
commit bf5c31ae03

View File

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