From d261bec1ec5a185b6e57f73e8589b07d5d5bc4e8 Mon Sep 17 00:00:00 2001
From: invincibledude <>
Date: Sun, 22 Jan 2023 16:14:28 +0300
Subject: [PATCH] Gen params paste improvement

---
 modules/generation_parameters_copypaste.py | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/modules/generation_parameters_copypaste.py b/modules/generation_parameters_copypaste.py
index 67e76e30..155ab350 100644
--- a/modules/generation_parameters_copypaste.py
+++ b/modules/generation_parameters_copypaste.py
@@ -252,13 +252,6 @@ Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 965400086, Size: 512x512, Model
         if line.startswith("Negative prompt:"):
             done_with_prompt = True
             line = line[16:].strip()
-
-        if line.startswith("Hires prompt:"):
-            res["Hires prompt"] = line[1:][:-1]
-
-        if line.startswith("Hires negative prompt:"):
-            res["Hires negative prompt"] = line[1:][:-1]
-
         if done_with_prompt:
             negative_prompt += ("" if negative_prompt == "" else "\n") + line
         else:
@@ -274,6 +267,10 @@ Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 965400086, Size: 512x512, Model
             res[k+"-2"] = m.group(2)
         else:
             res[k] = v
+        if k.startswith("Hires prompt:"):
+            res["Hires prompt"] = v[1:][:-1]
+        elif k.startswith("Hires negative prompt:"):
+            res["Hires negative prompt"] = v[1:][:-1]
 
     # Missing CLIP skip means it was set to 1 (the default)
     if "Clip skip" not in res: