hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-05 03:32:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Basic settings for realesrgan model selection.

Browse Source 
I don't like that you have to restart the app, but it works.
This commit is contained in:
d8ahazard
2022-09-21 17:23:38 -05:00
committed by AUTOMATIC1111
parent 4f434c8aea
commit c214c428bc
3 changed files with 72 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modules/shared.py
View File

@@ -13,6 +13,7 @@ import modules.styles
import modules.interrogate
import modules.memmon
import modules.sd_models
from modules.realesrgan_model_loader import get_realesrgan_model_names
sd_model_file = os.path.join(script_path, 'model.ckpt')
default_sd_model_file = sd_model_file
@@ -142,6 +143,10 @@ class Options:
"save_txt": OptionInfo(False, "Create a text file next to every image with generation parameters."),
"ESRGAN_tile": OptionInfo(192, "Tile size for ESRGAN upscalers. 0 = no tiling.", gr.Slider, {"minimum": 0, "maximum": 512, "step": 16}),
"ESRGAN_tile_overlap": OptionInfo(8, "Tile overlap, in pixels for ESRGAN upscalers. Low values = visible seam.", gr.Slider, {"minimum": 0, "maximum": 48, "step": 1}),
"realesrgan_enabled_models": OptionInfo(
["Real-ESRGAN 2x plus", "Real-ESRGAN 4x plus", "Real-ESRGAN 4x plus anime 6B"],
"Select which RealESRGAN models to show in the web UI. (Requires restart)",
gr.CheckboxGroup, {"choices": get_realesrgan_model_names()}),
"SWIN_tile": OptionInfo(192, "Tile size for all SwinIR.", gr.Slider, {"minimum": 16, "maximum": 512, "step": 16}),
"SWIN_tile_overlap": OptionInfo(8, "Tile overlap, in pixels for SwinIR. Low values = visible seam.", gr.Slider, {"minimum": 0, "maximum": 48, "step": 1}),
"ldsr_steps": OptionInfo(100, "LDSR processing steps. Lower = faster",