File size: 2,142 Bytes
c336648 |
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 |
try:
import smea_sampling
from smea_sampling import sample_euler_dy, sample_euler_smea_dy, sample_euler_negative, sample_euler_dy_negative
if smea_sampling.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):
"SMEA Samplers"
def show(self, is_img2img):
return False
def __init__(self):
if not smea_sampling.INITIALIZED:
samplers_smea = [
("Euler Dy", sample_euler_dy, ["k_euler_dy"], {}),
("Euler SMEA Dy", sample_euler_smea_dy, ["k_euler_smea_dy"], {}),
("Euler Negative", sample_euler_negative, ["k_euler_negative"], {}),
("Euler Negative Dy", sample_euler_dy_negative, ["k_euler_negative_dy"], {}),
]
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_euler_dy"] = ["s_churn", "s_tmin", "s_tmax", "s_noise"]
sampler_extra_params["sample_euler_smea_dy"] = ["s_churn", "s_tmin", "s_tmax", "s_noise"]
sampler_extra_params["sample_euler_negative"] = ["s_churn", "s_tmin", "s_tmax", "s_noise"]
sampler_extra_params["sample_euler_dy_negative"] = ["s_churn", "s_tmin", "s_tmax", "s_noise"]
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_sampling.INITIALIZED = True
except ImportError as _:
pass
|