refactor: use central flag for ControlNet image count

This commit is contained in:
Manuel Schmid
2024-01-29 14:27:51 +01:00
parent c3ab9f1f30
commit 20e53028a4
3 changed files with 4 additions and 2 deletions
+1 -1
View File
@@ -147,7 +147,7 @@ def worker():
metadata_scheme = MetadataScheme(args.pop()) if not args_manager.args.disable_metadata else MetadataScheme.FOOOCUS
cn_tasks = {x: [] for x in flags.ip_list}
for _ in range(4):
for _ in range(flags.controlnet_image_count):
cn_img = args.pop()
cn_stop = args.pop()
cn_weight = args.pop()