From 4078b82a8087a8d34a7bf85eecf7cd2a1c989ac2 Mon Sep 17 00:00:00 2001 From: Rhys Yang Date: Tue, 19 Sep 2023 03:12:29 +0800 Subject: [PATCH] Handle empty parameter --- nodes.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/nodes.py b/nodes.py index 10c14ce..e0f36a1 100644 --- a/nodes.py +++ b/nodes.py @@ -107,11 +107,11 @@ class SDPromptReader: output_to_terminal(MESSAGE["format_error"][1]) raise ValueError(MESSAGE["format_error"][1]) - seed = int(image_data.parameter.get("seed")) - steps = int(image_data.parameter.get("steps")) - cfg = float(image_data.parameter.get("cfg")) - width = int(image_data.width) - height = int(image_data.height) + seed = int(image_data.parameter.get("seed") or 0) + steps = int(image_data.parameter.get("steps") or 0) + cfg = float(image_data.parameter.get("cfg") or 0) + width = int(image_data.width or 0) + height = int(image_data.height or 0) output_to_terminal("Positive: \n" + image_data.positive) output_to_terminal("Negative: \n" + image_data.negative)