wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 23:00:51 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

robust patch for None return

Browse Source
This commit is contained in:
Dr.Lt.Data 2023-06-17 16:10:49 +09:00
parent 40402921d7
commit 7cde7443b0
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
execution.py
View File

@ -90,7 +90,7 @@ def get_output_data(obj, input_data_all):
results.append(r) results.append(r)
output = [] output = []
if len(results) > 0: if len(results) > 0 and results[0] is not None:
# check which outputs need concatenating # check which outputs need concatenating
output_is_list = [False] * len(results[0]) output_is_list = [False] * len(results[0])
if hasattr(obj, "OUTPUT_IS_LIST"): if hasattr(obj, "OUTPUT_IS_LIST"):

4
worklist_execution.py
View File

@ -110,7 +110,9 @@ def is_incomplete_input_slots(class_def, inputs, outputs):
if isinstance(input_data, list): if isinstance(input_data, list):
input_unique_id = input_data[0] input_unique_id = input_data[0]
if input_unique_id not in outputs or outputs[input_unique_id][input_data[1]] == [None]: if input_unique_id not in outputs or \
len(outputs[input_unique_id]) == 0 or \
outputs[input_unique_id][input_data[1]] == [None]:
return True return True
return False return False