File size: 1,093 Bytes
82ea528 |
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 |
import logging
from comfy.samplers import KSAMPLER
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
log = logging.getLogger(__name__)
from ..sampling.sampling_functions import get_rf_reverse_sample_fn
class MochiResamplerNode:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"eta": ("FLOAT", {"default": 0.9, "min": 0.0, "max": 30.0, "step": 0.01}),
"start_step": ("INT", {"default": 0, "min": 0}),
"end_step": ("INT", {"default": 10, "min": 0}),
"eta_trend": (['constant', 'linear_decrease', 'linear_increase'],),
"latents": ("LATENT", ),
}
}
RETURN_TYPES = ("SAMPLER",)
FUNCTION = "process"
CATEGORY = "MochiEdit"
def process(self, eta, start_step, end_step, eta_trend, latents):
latent_image = latents['samples']
sampler_fn = get_rf_reverse_sample_fn(latent_image, eta, start_step, end_step, eta_trend)
sampler = KSAMPLER(sampler_fn)
return (sampler,)
|