nodes/receyuki-comfyui-prompt-reader-node
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix the parsing error caused by the loss of model data #43

Browse Source
This commit is contained in:
Rhys Yang
2024-01-17 06:28:20 +08:00
parent d1e6940491
commit 27bcc7273e
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+11 -4
View File
@@ -159,18 +159,21 @@ class SDPromptReader:
raise ValueError(MESSAGE["format_error"][1]) raise ValueError(MESSAGE["format_error"][1])
seed = int( seed = int(
self.param_parser(image_data.parameter.get("seed"), parameter_index) self.param_parser(image_data.parameter.get("seed", 0), parameter_index)
or 0 or 0
) )
steps = int( steps = int(
self.param_parser(image_data.parameter.get("steps"), parameter_index) self.param_parser(image_data.parameter.get("steps", 0), parameter_index)
or 0 or 0
) )
cfg = float( cfg = float(
self.param_parser(image_data.parameter.get("cfg"), parameter_index) or 0 self.param_parser(image_data.parameter.get("cfg", 0), parameter_index)
or 0
) )
model = str( model = str(
self.param_parser(image_data.parameter.get("model"), parameter_index) self.param_parser(
image_data.parameter.get("model", ""), parameter_index
)
or "" or ""
) )
width = int(image_data.width or 0) width = int(image_data.width or 0)
@@ -204,7 +207,11 @@ class SDPromptReader:
@staticmethod @staticmethod
def param_parser(data: str, index: int): def param_parser(data: str, index: int):
try:
data_list = data.strip("()").split(",") data_list = data.strip("()").split(",")
except AttributeError:
return None
else:
return data_list[0] if len(data_list) == 1 else data_list[index] return data_list[0] if len(data_list) == 1 else data_list[index]
@staticmethod @staticmethod