lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-03 03:59:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1340 from DenOfEquity/fix-for-new-samplers

Browse Source 
Fix for new samplers
This commit is contained in:
Panchovix
2024-08-20 10:12:37 -04:00
committed by GitHub
parent 14ac95f908 c127e60cf0
commit f136f86fee
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/sd_samplers_kdiffusion.py
View File

@@ -27,6 +27,10 @@ samplers_k_diffusion = [
('DPM fast', 'sample_dpm_fast', ['k_dpm_fast'], {"uses_ensd": True}),
('DPM adaptive', 'sample_dpm_adaptive', ['k_dpm_ad'], {"uses_ensd": True}),
('Restart', sd_samplers_extra.restart_sampler, ['restart'], {'scheduler': 'karras', "second_order": True}),
('HeunPP2', 'sample_heunpp2', ['heunpp2'], {}),
('IPNDM', 'sample_ipndm', ['ipndm'], {}),
('IPNDM_V', 'sample_ipndm_v', ['ipndm_v'], {}),
('DEIS', 'sample_deis', ['deis'], {}),
]

4
modules_forge/alter_samplers.py
View File

@@ -19,8 +19,4 @@ def build_constructor(sampler_name):
samplers_data_alter = [
sd_samplers_common.SamplerData('DDPM', build_constructor(sampler_name='ddpm'), ['ddpm'], {}),
sd_samplers_common.SamplerData('HeunPP2', build_constructor(sampler_name='heunpp2'), ['heunpp2'], {}),
sd_samplers_common.SamplerData('IPNDM', build_constructor(sampler_name='ipndm'), ['ipndm'], {}),
sd_samplers_common.SamplerData('IPNDM_V', build_constructor(sampler_name='ipndm_v'), ['ipndm_v'], {}),
sd_samplers_common.SamplerData('DEIS', build_constructor(sampler_name='deis'), ['deis'], {}),
]