1580158
/
sd-webui-stable-horde-worker
mirror of https://github.com/sdwebui-w-horde/sd-webui-stable-horde-worker.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simplified and added samplers (#125)

master
Gabriel 2024-06-06 10:40:36 +02:00 committed by GitHub
parent ff9af8ade4
commit e206c36452
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
stable_horde/horde.py
View File

@ -222,67 +222,24 @@ class StableHorde:
samplers = [ samplers = [
SamplerData( SamplerData(
"Euler a Karras", name,
lambda model, funcname="sample_euler_ancestral": KDiffusionSampler( lambda model, fn=func: KDiffusionSampler(fn, model),
funcname, model [alias],
),
["k_euler_a_ka"],
{"scheduler": "karras"}, {"scheduler": "karras"},
), )
SamplerData( for name, func, alias in [
"Euler Karras", ("Euler a Karras", "sample_euler_ancestral", "k_euler_a_ka"),
lambda model, funcname="sample_euler": KDiffusionSampler( ("Euler Karras", "sample_euler", "k_euler_ka"),
funcname, model ("LMS Karras", "sample_lms", "k_lms_ka"),
), ("Heun Karras", "sample_heun", "k_heun_ka"),
["k_euler_ka"], ("DPM2 Karras", "sample_dpm_2", "k_dpm_2_ka"),
{"scheduler": "karras"}, ("DPM2 a Karras", "sample_dpm_2_ancestral", "k_dpm_2_a_ka"),
), ("DPM++ 2S a Karras", "sample_dpmpp_2s_ancestral", "k_dpmpp_2s_a_ka"),
SamplerData( ("DPM++ 2M Karras", "sample_dpmpp_2m", "k_dpmpp_2m_ka"),
"Heun Karras", ("DPM++ SDE Karras", "sample_dpmpp_sde", "k_dpmpp_sde_ka"),
lambda model, funcname="sample_heun": KDiffusionSampler( ("DPM fast Karras", "sample_dpm_fast", "k_dpm_fast_ka"),
funcname, model ("DPM adaptive Karras", "sample_dpm_adaptive", "k_dpm_ad_ka"),
), ]
["k_heun_ka"],
{"scheduler": "karras"},
),
SamplerData(
"DPM adaptive Karras",
lambda model, funcname="sample_dpm_adaptive": KDiffusionSampler(
funcname, model
),
["k_dpm_ad_ka"],
{"scheduler": "karras"},
),
SamplerData(
"DPM fast Karras",
lambda model, funcname="sample_dpm_fast": KDiffusionSampler(
funcname, model
),
["k_dpm_fast_ka"],
{"scheduler": "karras"},
),
SamplerData(
"LMS Karras",
lambda model, funcname="sample_lms": KDiffusionSampler(funcname, model),
["k_lms_ka"],
{"scheduler": "karras"},
),
SamplerData(
"DPM++ SDE Karras",
lambda model, funcname="sample_dpmpp_sde": KDiffusionSampler(
funcname, model
),
["k_dpmpp_sde_ka"],
{"scheduler": "karras"},
),
SamplerData(
"DPM++ 2S a Karras",
lambda model, funcname="sample_dpmpp_2s_ancestral": KDiffusionSampler(
funcname, model
),
["k_dpmpp_2s_a_ka"],
{"scheduler": "karras"},
),
] ]
sd_samplers.samplers.extend(samplers) sd_samplers.samplers.extend(samplers)
sd_samplers.samplers_for_img2img.extend(samplers) sd_samplers.samplers_for_img2img.extend(samplers)