From 2cf23099ebb81832a27c8016f14062885f5a9c98 Mon Sep 17 00:00:00 2001
From: w-e-w <40751091+w-e-w@users.noreply.github.com>
Date: Thu, 18 Jan 2024 04:44:21 +0900
Subject: fix console total progress bar when using txt2img_upscale

add p.txt2img_upscale as indicator
---
 modules/txt2img.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'modules/txt2img.py')

diff --git a/modules/txt2img.py b/modules/txt2img.py
index 92a160d6..4efcb4c3 100644
--- a/modules/txt2img.py
+++ b/modules/txt2img.py
@@ -64,6 +64,7 @@ def txt2img_upscale(id_task: str, request: gr.Request, gallery, gallery_index, g
     p.enable_hr = True
     p.batch_size = 1
     p.n_iter = 1
+    p.txt2img_upscale = True
 
     geninfo = json.loads(generation_info)
 
-- 
cgit v1.2.3