Mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-06-07 21:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

use re instead of hardcoding

Browse Source 
Now supports all natively provided upscaler as well
This commit is contained in:
Haoming 2024-03-14 09:55:14 +08:00
parent d18eb10ecd
commit fd71b761ff
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
modules/ui_postprocessing.py
View File

@ -5,6 +5,9 @@ from modules.ui_components import ResizeHandleRow
def hook_scale_update(inputs): def hook_scale_update(inputs):
import re
pattern = r'(\d)[xX]|[xX](\d)'
resize = upscaler = None resize = upscaler = None
for script in inputs: for script in inputs:
if script.label == "Resize": if script.label == "Resize":
@ -15,13 +18,16 @@ def hook_scale_update(inputs):
break break
def update_scale(upscaler: str, slider: float): def update_scale(upscaler: str, slider: float):
if upscaler[1] in ('x', 'X'): match = re.search(pattern, upscaler)
try:
scale = int(upscaler[0])
return gr.update(value=scale)
except ValueError:
return gr.update(value=slider)
if match:
if match.group(1):
return gr.update(value=int(match.group(1)))
else:
return gr.update(value=int(match.group(2)))
else:
return gr.update(value=slider) return gr.update(value=slider)
if resize and upscaler: if resize and upscaler: