aboutsummaryrefslogtreecommitdiffstats
path: root/modules/shared_options.py
diff options
context:
space:
mode:
authorn0kovo <n0kovo@riseup.net>2024-01-18 23:10:02 +0000
committern0kovo <n0kovo@riseup.net>2024-01-18 23:10:02 +0000
commita97147bc8a43ade7c18bb755f0cfac111fc1a619 (patch)
treeae59f7af8d7ca0c6b4be59bec4ab09ec1ae3ddb9 /modules/shared_options.py
parentcb5b335acddd126d4f6c990982816c06beb0d6ae (diff)
downloadstable-diffusion-webui-gfx803-a97147bc8a43ade7c18bb755f0cfac111fc1a619.tar.gz
stable-diffusion-webui-gfx803-a97147bc8a43ade7c18bb755f0cfac111fc1a619.tar.bz2
stable-diffusion-webui-gfx803-a97147bc8a43ade7c18bb755f0cfac111fc1a619.zip
Add support for DAT upscaler models
Diffstat (limited to 'modules/shared_options.py')
-rw-r--r--modules/shared_options.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/shared_options.py b/modules/shared_options.py
index 63488f4e..48a206ce 100644
--- a/modules/shared_options.py
+++ b/modules/shared_options.py
@@ -97,6 +97,9 @@ options_templates.update(options_section(('upscaling', "Upscaling", "postprocess
"ESRGAN_tile": OptionInfo(192, "Tile size for ESRGAN upscalers.", gr.Slider, {"minimum": 0, "maximum": 512, "step": 16}).info("0 = no tiling"),
"ESRGAN_tile_overlap": OptionInfo(8, "Tile overlap for ESRGAN upscalers.", gr.Slider, {"minimum": 0, "maximum": 48, "step": 1}).info("Low values = visible seam"),
"realesrgan_enabled_models": OptionInfo(["R-ESRGAN 4x+", "R-ESRGAN 4x+ Anime6B"], "Select which Real-ESRGAN models to show in the web UI.", gr.CheckboxGroup, lambda: {"choices": shared_items.realesrgan_models_names()}),
+ "dat_enabled_models": OptionInfo(["R-ESRGAN 4x+", "R-ESRGAN 4x+ Anime6B"], "Select which DAT models to show in the web UI.", gr.CheckboxGroup, lambda: {"choices": shared_items.dat_models_names()}),
+ "DAT_tile": OptionInfo(192, "Tile size for DAT upscalers.", gr.Slider, {"minimum": 0, "maximum": 512, "step": 16}).info("0 = no tiling"),
+ "DAT_tile_overlap": OptionInfo(8, "Tile overlap for DAT upscalers.", gr.Slider, {"minimum": 0, "maximum": 48, "step": 1}).info("Low values = visible seam"),
"upscaler_for_img2img": OptionInfo(None, "Upscaler for img2img", gr.Dropdown, lambda: {"choices": [x.name for x in shared.sd_upscalers]}),
}))