xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 22:57:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

use control_weights in HunyuanDit Controlnet

Browse Source
This commit is contained in:
CrazyBoyM 2024-08-08 00:34:09 +08:00
parent e724bd542c
commit 4d55f2e28c
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/ldm/hydit/controlnet.py
View File

@ -337,6 +337,12 @@ class HunYuanControlNet(nn.Module):
x = block(x, c, text_states, freqs_cis_img) x = block(x, c, text_states, freqs_cis_img)
controls.append(self.after_proj_list[layer](x)) # zero linear for output controls.append(self.after_proj_list[layer](x)) # zero linear for output
controls = [control * control_weight for control in controls] control_weights = [1.0 * (control_weight ** float(19 - i)) for i in range(19)]
assert len(control_weight) == len(
controls
), "control_weights and controls should have the same length"
controls = [
control * weight for control, weight in zip(controls, control_weights)
]
return {"output": controls} return {"output": controls}