Merge pull request #11047 from AUTOMATIC1111/parse_generation_parameters_with_error

handles exception when parsing generation parameters from png info

1 files changed, 12 insertions, 9 deletions

diff --git a/modules/generation_parameters_copypaste.py b/modules/generation_parameters_copypaste.py
index 1d02ffae..a638f912 100644
--- a/modules/generation_parameters_copypaste.py
+++ b/modules/generation_parameters_copypaste.py
@@ -277,15 +277,18 @@ Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 965400086, Size: 512x512, Model
     res["Negative prompt"] = negative_prompt

 

     for k, v in re_param.findall(lastline):

-        if v[0] == '"' and v[-1] == '"':

-            v = unquote(v)

-

-        m = re_imagesize.match(v)

-        if m is not None:

-            res[f"{k}-1"] = m.group(1)

-            res[f"{k}-2"] = m.group(2)

-        else:

-            res[k] = v

+        try:

+            if v[0] == '"' and v[-1] == '"':

+                v = unquote(v)

+

+            m = re_imagesize.match(v)

+            if m is not None:

+                res[f"{k}-1"] = m.group(1)

+                res[f"{k}-2"] = m.group(2)

+            else:

+                res[k] = v

+        except Exception:

+            print(f"Error parsing \"{k}: {v}\"")

 

     # Missing CLIP skip means it was set to 1 (the default)

     if "Clip skip" not in res: