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. |
Problems:
- The node's progress bar in the UI will not accurately reflect the sampling progress. Use the console for accurate progress.