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

2024-01-17 06:28:20 +08:00
parent d1e6940491
commit 27bcc7273e
1 changed files with 13 additions and 6 deletions
@@ -159,18 +159,21 @@ class SDPromptReader:
                raise ValueError(MESSAGE["format_error"][1])

            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
            )
            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
            )
            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(
-                self.param_parser(image_data.parameter.get("model"), parameter_index)
+                self.param_parser(
+                    image_data.parameter.get("model", ""), parameter_index
+                )
                or ""
            )
            width = int(image_data.width or 0)
@@ -204,7 +207,11 @@ class SDPromptReader:

    @staticmethod
    def param_parser(data: str, index: int):
+        try:
            data_list = data.strip("()").split(",")
+        except AttributeError:
+            return None
+        else:
            return data_list[0] if len(data_list) == 1 else data_list[index]

    @staticmethod