This commit is contained in:
CrazyBoyM 2024-08-08 00:36:44 +08:00
parent 4d55f2e28c
commit c0a0400d99

View File

@ -338,7 +338,7 @@ class HunYuanControlNet(nn.Module):
controls.append(self.after_proj_list[layer](x)) # zero linear for output controls.append(self.after_proj_list[layer](x)) # zero linear for output
control_weights = [1.0 * (control_weight ** float(19 - i)) for i in range(19)] control_weights = [1.0 * (control_weight ** float(19 - i)) for i in range(19)]
assert len(control_weight) == len( assert len(control_weights) == len(
controls controls
), "control_weights and controls should have the same length" ), "control_weights and controls should have the same length"
controls = [ controls = [