uses autos new regex, checks len of re_param

1 files changed, 2 insertions, 2 deletions

diff --git a/modules/generation_parameters_copypaste.py b/modules/generation_parameters_copypaste.py
index 13d0874d..53f1a865 100644
--- a/modules/generation_parameters_copypaste.py
+++ b/modules/generation_parameters_copypaste.py
@@ -11,7 +11,7 @@ from modules import shared, ui_tempdir, script_callbacks
 import tempfile

 from PIL import Image

 

-re_param_code = r'\s*([\w ]+):\s*(\"[^\"]*\"|[^,]+)'

+re_param_code = r'\s*([\w ]+):\s*("(?:\\"[^,]|\\"|\\|[^\"])+"|[^,]*)(?:,|$)'

 re_param = re.compile(re_param_code)

 re_imagesize = re.compile(r"^(\d+)x(\d+)$")

 re_hypernet_hash = re.compile("\(([0-9a-f]+)\)$")

@@ -242,7 +242,7 @@ Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 965400086, Size: 512x512, Model
     done_with_prompt = False

 

     *lines, lastline = x.strip().split("\n")

-    if not re_param.match(lastline):

+    if len(re_param.findall(lastline)) < 3:

         lines.append(lastline)

         lastline = ''