File size: 2,599 Bytes
899ba56 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
try:
import smea
from smea import sample_spawner_smea, sample_spawner_smea_beta, sample_spawner_smea_dyn_beta, sample_spawner_smea_dyn_beta1, sample_spawner_rk2_smea_d_clamp, sample_spawner_smea_dyn_exp
if smea.BACKEND == "WebUI":
from modules import scripts, sd_samplers_common, sd_samplers
from modules.sd_samplers_kdiffusion import sampler_extra_params, KDiffusionSampler
class SMEA(scripts.Script):
def title(self):
"Spawner SMEA Samplers"
def show(self, is_img2img):
return False
def __init__(self):
if not smea.INITIALIZED:
samplers_smea = [
("Spawner SMEA", sample_spawner_smea, ["k_spawner_smea"], {"uses_ensd": True}),
("Spawner SMEA (beta)", sample_spawner_smea_beta, ["k_spawner_smea_beta"], {"uses_ensd": True}),
("Spawner SMEA Dyn (beta)", sample_spawner_smea_dyn_beta, ["k_spawner_smea_dyn_beta"], {"uses_ensd": True}),
("Spawner SMEA Dyn (beta1)", sample_spawner_smea_dyn_beta1, ["k_spawner_smea_dyn_beta1"], {"uses_ensd": True}),
("Spawner SMEA Dyn exp", sample_spawner_smea_dyn_exp, ["k_spawner_smea_dyn_exp"], {}),
('Spawner rk2 smea d clamp', sample_spawner_rk2_smea_d_clamp, ['spawner_rk2_smea_d_clamp'], {}),
]
samplers_data_smea = [
sd_samplers_common.SamplerData(label, lambda model, funcname=funcname: KDiffusionSampler(funcname, model), aliases, options)
for label, funcname, aliases, options in samplers_smea
if callable(funcname)
]
sampler_extra_params["sample_spawner_smea"] = ['s_noise', 'eta']
sampler_extra_params["sample_spawner_smea_beta"] = ['eta', 's_noise', 'beta']
sampler_extra_params["sample_spawner_smea_dyn_beta"] = ['eta_start', 'eta_end', 'eta_exponent', 's_noise', 'beta', 'sigma_max_for_dyn_eta']
sampler_extra_params["sample_spawner_smea_dyn_beta1"] = ['eta_start', 'eta_end', 'eta_exponent', 's_noise', 'beta', 'sigma_max_for_dyn_eta']
sd_samplers.all_samplers.extend(samplers_data_smea)
sd_samplers.all_samplers_map = {x.name: x for x in sd_samplers.all_samplers}
sd_samplers.set_samplers()
smea.INITIALIZED = True
except ImportError as _:
pass
|