ComfyUI_restart_sampling

Unofficial ComfyUI nodes for restart sampling based on the paper "Restart Sampling for Improving Generative Processes" [paper] [repo]

Installation

Enter the following command from the commandline starting in ComfyUI/custom_nodes/

git clone https://github.com/ssitu/ComfyUI_restart_sampling

Usage

Nodes can be found in the node menu under sampling:

Node	Image	Description
KSampler With Restarts		Has all the inputs of a KSampler, but with an added string widget for configuring the Restart segments and a widget for the scheduler for the Restart segments. Not all samplers and schedulers from KSampler are currently supported. The format for segments is a sequence of comma separated arrays of {[N_{\textrm{Restart}}, K, t_{\textrm{min}}, t_{\textrm{max}}]}. For example, [4, 1, 19.35, 40.79], [4, 1, 1.09, 1.92], [4, 5, 0.59, 1.09], [4, 5, 0.30, 0.59], [6, 6, 0.06, 0.30] would be a valid sequence. Each segment will add (N_{\textrm{Restart}} - 1) \cdot K steps to the sampling process. For more information on the Restart parameters, refer to the paper. The restart_scheduler is used as the scheduler for the backwards process during restart intervals. The researchers used the karras scheduler for their experiments.

---