Fixes to make controlnet type models work on qwen edit and kontext. (#9581)

This commit is contained in:
comfyanonymous
2025-08-27 12:26:28 -07:00
committed by GitHub
parent b20ba1f27c
commit b5ac6ed7ce
3 changed files with 8 additions and 6 deletions

View File

@@ -459,7 +459,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)