diff --git a/paste_this_to_webui_scripts_folder/prompt_gallery.py b/paste_this_to_webui_scripts_folder/prompt_gallery.py index 72bdfcc..3ab22ae 100644 --- a/paste_this_to_webui_scripts_folder/prompt_gallery.py +++ b/paste_this_to_webui_scripts_folder/prompt_gallery.py @@ -158,7 +158,10 @@ map_sampler_to_idx = { 'DPM2 Karras': 9, 'DPM2 a Karras': 10, 'DDIM': 11, - 'PLMS': 12} + 'PLMS': 12, + 'DPM++ 2S a Karras': 13, + 'DPM++ 2M Karras': 14, + 'DPM++ SDE Karras': 15} map_keys = { "value": "prompt", @@ -712,4 +715,4 @@ class Script(scripts.Script): rawDict = {} shared.opts = backup - return Processed(p, images, p.seed, "") \ No newline at end of file + return Processed(p, images, p.seed, "")