diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index fe2d0fb25d8111758cc4fc83daf2b54f6af57aab..666ee1ee207f89ed58b73a614af696178a4e44a8 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -38,9 +38,9 @@ samplers = [ samplers_for_img2img = [x for x in samplers if x.name != 'PLMS'] sampler_extra_params = { - 'sample_euler':['s_churn','s_tmin','s_noise'], - 'sample_heun' :['s_churn','s_tmin','s_noise'], - 'sample_dpm_2':['s_churn','s_tmin','s_noise'], + 'sample_euler':['s_churn','s_tmin','s_tmax','s_noise'], + 'sample_heun' :['s_churn','s_tmin','s_tmax','s_noise'], + 'sample_dpm_2':['s_churn','s_tmin','s_tmax','s_noise'], } def setup_img2img_steps(p, steps=None):