Skip to content

Commit f6274c0

Browse files
Fix issue with batch_size > 1 on some models. (Comfy-Org#12892)
1 parent 4f4f865 commit f6274c0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

comfy/ldm/flux/layers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -144,9 +144,9 @@ def apply_mod(tensor, m_mult, m_add=None, modulation_dims=None):
144144
return tensor * m_mult
145145
else:
146146
for d in modulation_dims:
147-
tensor[:, d[0]:d[1]] *= m_mult[:, d[2]]
147+
tensor[:, d[0]:d[1]] *= m_mult[:, d[2]:d[2] + 1]
148148
if m_add is not None:
149-
tensor[:, d[0]:d[1]] += m_add[:, d[2]]
149+
tensor[:, d[0]:d[1]] += m_add[:, d[2]:d[2] + 1]
150150
return tensor
151151

152152

0 commit comments

Comments
 (0)