@@ -295,6 +295,23 @@ def get_sampler(self, eta, s_noise, r, noise_device):
295295 sampler = comfy .samplers .ksampler (sampler_name , {"eta" : eta , "s_noise" : s_noise , "r" : r })
296296 return (sampler , )
297297
298+ class SamplerDPMPP_2S_Ancestral :
299+ @classmethod
300+ def INPUT_TYPES (s ):
301+ return {"required" :
302+ {"eta" : ("FLOAT" , {"default" : 1.0 , "min" : 0.0 , "max" : 100.0 , "step" :0.01 , "round" : False }),
303+ "s_noise" : ("FLOAT" , {"default" : 1.0 , "min" : 0.0 , "max" : 100.0 , "step" :0.01 , "round" : False }),
304+ }
305+ }
306+ RETURN_TYPES = ("SAMPLER" ,)
307+ CATEGORY = "sampling/custom_sampling/samplers"
308+
309+ FUNCTION = "get_sampler"
310+
311+ def get_sampler (self , eta , s_noise ):
312+ sampler = comfy .samplers .ksampler ("dpmpp_2s_ancestral" , {"eta" : eta , "s_noise" : s_noise })
313+ return (sampler , )
314+
298315class SamplerEulerAncestral :
299316 @classmethod
300317 def INPUT_TYPES (s ):
@@ -666,6 +683,7 @@ def add_noise(self, model, noise, sigmas, latent_image):
666683 "SamplerDPMPP_3M_SDE" : SamplerDPMPP_3M_SDE ,
667684 "SamplerDPMPP_2M_SDE" : SamplerDPMPP_2M_SDE ,
668685 "SamplerDPMPP_SDE" : SamplerDPMPP_SDE ,
686+ "SamplerDPMPP_2S_Ancestral" : SamplerDPMPP_2S_Ancestral ,
669687 "SamplerDPMAdaptative" : SamplerDPMAdaptative ,
670688 "SplitSigmas" : SplitSigmas ,
671689 "SplitSigmasDenoise" : SplitSigmasDenoise ,
@@ -682,4 +700,4 @@ def add_noise(self, model, noise, sigmas, latent_image):
682700
683701NODE_DISPLAY_NAME_MAPPINGS = {
684702 "SamplerEulerAncestralCFGPP" : "SamplerEulerAncestralCFG++" ,
685- }
703+ }
0 commit comments