aboutsummaryrefslogtreecommitdiff
path: root/modules/ui_toprow.py
diff options
context:
space:
mode:
authorKohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>2023-12-03 10:54:54 +0800
committerKohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>2023-12-03 10:54:54 +0800
commit9a15ae2a92e55d614fe515cd0a104d90b854b23f (patch)
tree7977ea1ea27cfc1d21e652433f8bbc0faec0ddc9 /modules/ui_toprow.py
parent50a21cb09fe3e9ea2d4fe058e0484e192c8a86e3 (diff)
parentac02216e540cd581f9169c6c791e55721e3117b0 (diff)
Merge branch 'dev' into test-fp8
Diffstat (limited to 'modules/ui_toprow.py')
-rw-r--r--modules/ui_toprow.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/modules/ui_toprow.py b/modules/ui_toprow.py
index 985b5a2d..88838f97 100644
--- a/modules/ui_toprow.py
+++ b/modules/ui_toprow.py
@@ -34,8 +34,10 @@ class Toprow:
submit_box = None
- def __init__(self, is_img2img, is_compact=False):
- id_part = "img2img" if is_img2img else "txt2img"
+ def __init__(self, is_img2img, is_compact=False, id_part=None):
+ if id_part is None:
+ id_part = "img2img" if is_img2img else "txt2img"
+
self.id_part = id_part
self.is_img2img = is_img2img
self.is_compact = is_compact