Added support for specifying which step iteration to start using

each ControlNet, and which step to end using each controlnet (specified as fraction of total steps)
This commit is contained in:
user1
2023-05-12 04:01:35 -07:00
committed by Kent Keirsey
parent 11fc7e40a5
commit 754acec92f
2 changed files with 16 additions and 13 deletions

View File

@ -312,7 +312,9 @@ class TextToLatentsInvocation(BaseInvocation):
)
control_item = ControlNetData(model=control_model,
image_tensor=control_image,
weight=control_info.control_weight)
weight=control_info.control_weight,
begin_step_percent=control_info.begin_step_percent,
end_step_percent=control_info.end_step_percent)
control_data.append(control_item)
# multi_control = MultiControlNetModel(control_models)
# model.control_model = multi_control