50 Commits
Author SHA1 Message Date
blepping 25d8be7160 Support flow models 2025-12-05 13:02:27 -07:00
blepping 985fc1f3e8 Tiny cleanup 2024-05-04 05:18:57 -06:00
blepping b00812334b Discard penultimate sigma per default behavior when sampler is passed by name 2024-05-04 04:06:58 -06:00
blepping d17907e09d Update PlanItem comment for s_min, s_max properties, add n_restart property for consistency 2024-05-03 14:01:12 -06:00
blepping 725d3aeb58 Update documentation for restart sampler/scheduler changes
Fix denoise/step range calculation for really reals this time, I hope

Use the normal scheduler list for non-restart schedulers in nodes

Revert limiting s_max to model sigma_max

Add sgm_uniform to restart schedulers list (just normal with sgm=True)
2024-04-23 13:30:12 -06:00
blepping 83470d59e0 Fix plan denoise/start/end step calculation, hopefully 2024-04-21 16:30:32 -06:00
blepping 7ef2f56798 Fix total steps calculation in actual restart sampler 2024-04-21 05:12:30 -06:00
blepping a36442dd4f Cleanups, add some comments 2024-04-21 04:56:59 -06:00
blepping 1b65c60e0c Simplify converting plan to sigmas 2024-04-21 03:55:52 -06:00
blepping f5c7a877e5 Refactor and simplify 2024-04-19 05:58:05 -06:00
blepping 471b5972c9 only generate restart segment down to model sigma_min
merge restart segments more aggressively
2024-04-18 10:56:55 -06:00
blepping 20dc195ca9 Force last sigma in restart segment to t_min + other stuff 2024-04-17 14:05:13 -06:00
blepping bda63be15c The struggle! 2024-04-17 09:35:44 -06:00
blepping e2dcfd4091 Refactor plan recovery, make plan sampling function reusable 2024-04-16 18:59:07 -06:00
blepping 3752a2daf9 Make restart a sampler, add node to generate sigmas: phase 1 2024-04-15 10:35:00 -06:00
blepping a73142188a Only increment steps/delegate to callback when the sigma changes 2024-04-03 20:03:39 -06:00
blepping ae32ce995a Code formatting + clean up some lints 2024-04-01 12:40:09 -06:00
blepping 9c65214933 Attempt to get a1111 mode t_max calculation correct
Improve sigmas output when dumping restart plan
2024-04-01 03:34:24 -06:00
blepping f890b8bf62 Use correct value for a1111 mode t_max 2024-03-31 16:04:15 -06:00
blepping 17193ecdbf Minor cleanups + add a few more comments
Allow passing sigmas to main restart_sampling function
2024-03-28 02:00:34 -06:00
blepping e2e25e7a28 Add some comments describing what's going on in plan generation and execution
Allow specifying segments "default" to use the default segments

Allow specifying segments "a1111" to calculate segments like A1111

Allow setting environment variable COMFYUI_VERBOSE_RESTART_SAMPLING=1 to get some debug info

Add chunked_mode to samplers (except for simple which will use the default of True)

Documentation updates
2024-03-27 05:45:06 -06:00
blepping bbbddbd7cb Remove unused noise_multiplier param 2024-03-25 06:00:09 -06:00
blepping 67d4b62235 Refactor plan handling 2024-03-25 05:45:48 -06:00
blepping 12c0ca1044 Simplify total steps logic in wrapper 2024-03-22 14:51:20 -06:00
blepping b172908ac7 Implement chunked restart sampling 2024-03-22 14:45:04 -06:00
blepping 33ba61bb78 Make custom restart sampler with custom noise a separate node 2024-03-22 11:47:08 -06:00
blepping 22f4ee04e5 Hack to allow setting custom noise 2024-03-22 04:20:41 -06:00
blepping b5ef71262f Remove redundant seg assignment in restart wrapper 2024-03-17 13:17:25 -06:00
blepping eebbd9cf21 Allow specify t_min/max as timesteps and percentages in Restart segments 2024-03-17 13:11:00 -06:00
blepping 1fb1b03261 Fix setting denoise 2024-03-16 03:49:52 -06:00
blepping 20efcba391 Fix assuming first/last steps would be set for non-custom restart samplers 2024-03-15 19:20:58 -06:00
blepping 6ddbaf02f4 Add KRestartSamplerCustom + other cleanups 2024-03-14 05:41:27 -06:00
ssit 5872914f14 ddim has been redirected to ksampler euler
Fix for comfyui commits 1777b54 and a268a57
2023-11-02 22:23:20 -04:00
ssitandshiertier 704196b070 Change match to ifs for older python combatibility
Co-Authored-By: shiertier <90256303+shiertier@users.noreply.github.com>
2023-10-21 18:04:58 -04:00
ssit bfa396b1da Fix error with latest comfyui 2023-09-28 12:25:02 -04:00
ssit beb9dd2725 Ignore segments outside of the range of sigmas 2023-09-10 22:32:02 -04:00
ssit ebedccd454 Ensure that N and K are integers 2023-09-10 20:17:49 -04:00
ssit 8f5ef9b6ed Fix ddim sampler 2023-09-10 12:39:01 -04:00
ssit c3fec18633 Add advanced node, remove DPM++ 3m SDE 2023-09-10 11:05:46 -04:00
ssit 2098f265a4 Added support for ddim sampler 2023-07-08 23:07:37 -04:00
ssit 446cdc3f19 Added support for ddim_uniform scheduler
Also renamed schedulers.py to be more specific.
2023-07-05 12:39:31 -04:00
ssit 1053560d93 Added support for the simple scheduler 2023-07-04 12:13:12 -04:00
ssit a80ef83a84 Added support for the normal scheduler 2023-07-03 20:18:59 -04:00
ssit e4508f435a Fix wrong step count with redundant segments 2023-07-02 22:20:28 -04:00
ssit 0198efc64f Fixed the node ui progress bar 2023-07-02 13:13:40 -04:00
ssit 2788fea78d Add exponential scheduler and remove print statement 2023-06-30 16:50:26 -04:00
ssit 7e8188be68 Added proper progress bar and widget for restart scheduler
Also fixed the wrapper for other samplers that don't have the same parameters as heun.
2023-06-30 16:11:30 -04:00
ssit b4e7e042e0 Fixed sampler not being restored if interrupted 2023-06-30 15:22:19 -04:00
ssit f18f401051 Added text widget for configuring restart segments 2023-06-30 01:40:51 -04:00
ssit e031a87328 Node is functional. Hardcoding restarts for now 2023-06-29 19:48:19 -04:00