Merge branch 'master' into attention-select

This commit is contained in:
Jedrzej Kosinski
2025-08-29 23:35:38 -07:00
15 changed files with 253 additions and 105 deletions

View File

@@ -463,7 +463,7 @@ class QwenImageTransformer2DModel(nn.Module):
if i < len(control_i):
add = control_i[i]
if add is not None:
hidden_states += add
hidden_states[:, :add.shape[1]] += add
hidden_states = self.norm_out(hidden_states, temb)
hidden_states = self.proj_out(hidden_states)