diff options
| author | KyuSeok Jung <wjdrbtjr495@gmail.com> | 2022-11-05 18:08:45 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-05 18:08:45 +0000 |
| commit | 9b7289c3493ba1f42aa5e7657d385124fec1f112 (patch) | |
| tree | 37d86b581d265e06df7112e7a87f514de4702bc6 /modules/sd_samplers.py | |
| parent | 45b65e87e0ef64b3e457f7d20c62d591cdcd0e7b (diff) | |
| parent | b08698a09a257365238e43cc9023ce7cf9af73c4 (diff) | |
| download | stable-diffusion-webui-gfx803-9b7289c3493ba1f42aa5e7657d385124fec1f112.tar.gz stable-diffusion-webui-gfx803-9b7289c3493ba1f42aa5e7657d385124fec1f112.tar.bz2 stable-diffusion-webui-gfx803-9b7289c3493ba1f42aa5e7657d385124fec1f112.zip | |
Merge branch 'master' into master
Diffstat (limited to 'modules/sd_samplers.py')
| -rw-r--r-- | modules/sd_samplers.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 44d4c189..783992d2 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -24,11 +24,15 @@ samplers_k_diffusion = [ ('Heun', 'sample_heun', ['k_heun'], {}),
('DPM2', 'sample_dpm_2', ['k_dpm_2'], {}),
('DPM2 a', 'sample_dpm_2_ancestral', ['k_dpm_2_a'], {}),
+ ('DPM++ 2S a', 'sample_dpmpp_2s_ancestral', ['k_dpmpp_2s_a'], {}),
+ ('DPM++ 2M', 'sample_dpmpp_2m', ['k_dpmpp_2m'], {}),
('DPM fast', 'sample_dpm_fast', ['k_dpm_fast'], {}),
('DPM adaptive', 'sample_dpm_adaptive', ['k_dpm_ad'], {}),
('LMS Karras', 'sample_lms', ['k_lms_ka'], {'scheduler': 'karras'}),
('DPM2 Karras', 'sample_dpm_2', ['k_dpm_2_ka'], {'scheduler': 'karras'}),
('DPM2 a Karras', 'sample_dpm_2_ancestral', ['k_dpm_2_a_ka'], {'scheduler': 'karras'}),
+ ('DPM++ 2S a Karras', 'sample_dpmpp_2s_ancestral', ['k_dpmpp_2s_a_ka'], {'scheduler': 'karras'}),
+ ('DPM++ 2M Karras', 'sample_dpmpp_2m', ['k_dpmpp_2m_ka'], {'scheduler': 'karras'}),
]
samplers_data_k_diffusion = [
@@ -93,8 +97,8 @@ def single_sample_to_image(sample): return Image.fromarray(x_sample)
-def sample_to_image(samples):
- return single_sample_to_image(samples[0])
+def sample_to_image(samples, index=0):
+ return single_sample_to_image(samples[index])
def samples_to_image_grid(samples):
|