File size: 811 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 |
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_forward_sample_fn
class MochiUnsamplerNode:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}),
"gamma": ("FLOAT", {"default": 0.0, "min": 0.0, "max": 30.0, "step": 0.01}),
}
}
RETURN_TYPES = ("SAMPLER",)
FUNCTION = "process"
CATEGORY = "MochiEdit"
def process(self, seed, gamma):
sampler_fn = get_rf_forward_sample_fn(gamma, seed)
sampler = KSAMPLER(sampler_fn)
return (sampler,)
|