wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-15 03:27:24 +08:00
Code Issues Actions 25 Packages Projects Releases Wiki Activity

Sort combinatorial changes by input name

Browse Source
This commit is contained in:
space-nuko 2023-05-14 18:06:11 -05:00
parent ba6d6de2ae
commit b185b9cac7
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
execution.py
View File

@ -14,6 +14,7 @@ import nodes
import comfy.model_management import comfy.model_management
def get_input_data_batches(input_data_all): def get_input_data_batches(input_data_all):
"""Given input data that can contain combinatorial input values, returns all """Given input data that can contain combinatorial input values, returns all
the possible batches that can be made by combining the different input the possible batches that can be made by combining the different input
@ -22,8 +23,14 @@ def get_input_data_batches(input_data_all):
input_to_index = {} input_to_index = {}
index_to_values = [] index_to_values = []
# Sort by input name first so the order which batch inputs are applied can
# be easily calculated (node execution first, then alphabetical input name
# second)
sorted_input_names = sorted(input_data_all.keys())
i = 0 i = 0
for input_name, value in input_data_all.items(): for input_name in sorted_input_names:
value = input_data_all[input_name]
if isinstance(value, dict) and "combinatorial" in value: if isinstance(value, dict) and "combinatorial" in value:
input_to_index[input_name] = i input_to_index[input_name] = i
index_to_values.append(value["values"]) index_to_values.append(value["values"])