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 63d248622c
commit f613c073c1
3 changed files with 46 additions and 37 deletions

View File

@ -317,7 +317,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