65 Commits

Author SHA1 Message Date
chaObserv
e0db519c74 Add the denoising step to several samplers (#8780) 2025-07-03 19:20:53 -04:00
chaObserv
c64b35db3b Migrate ER-SDE from VE to VP algorithm and add its sampler node (#8744)
Apply alpha scaling in the algorithm for reverse-time SDE and add custom ER-SDE sampler node for other solver types (SDE, ODE).
2025-07-01 02:38:52 -04:00
chaObserv
b694938145 Singlestep DPM++ SDE for RF (#8627)
Refactor the algorithm, and apply alpha scaling.
2025-06-24 14:59:09 -04:00
chaObserv
058f21cd31 Multistep DPM++ SDE samplers for RF (#8541)
Include alpha in sampling and minor refactoring
2025-06-16 14:47:10 -04:00
chaObserv
dd027ae830 Generalize SEEDS samplers (#8529)
Restore VP algorithm for RF and refactor noise_coeffs and half-logSNR calculations
2025-06-14 16:58:16 -04:00
Pam
f459314eba Fix res_multistep_ancestral sampler (#8030) 2025-05-09 20:14:13 -04:00
chaObserv
4cc5ea895e CFG++ for gradient estimation sampler (#7809) 2025-04-28 13:51:35 -04:00
chaObserv
c4116ffaea Add SEEDS (stage 2 & 3 DP) sampler (#7580)
* Add seeds stage 2 & 3 (DP) sampler

* Change the name to SEEDS in comment
2025-04-12 18:36:08 -04:00
chaObserv
193e847c21 Guard the edge cases of noise term in er_sde (#7265) 2025-03-16 06:02:25 -04:00
chaObserv
bca2fadabc Ensure the extra_args in dpmpp sde series (#7204) 2025-03-12 17:28:59 -04:00
chaObserv
2e11c2bf43 Add er_sde sampler (#7187) 2025-03-12 02:42:37 -04:00
Pam
a1cd544ace res_multistep: Fix cfgpp and add ancestral samplers (#6731) 2025-02-08 19:39:58 -05:00
chaObserv
4d5a8e8525 Add gradient estimation sampler (#6554) 2025-01-22 05:29:40 -05:00
Sergii Dymchenko
441dfbf8d9 Use torch.special.expm1 (#6388)
* Use `torch.special.expm1`

This function provides greater precision than `exp(x) - 1` for small values of `x`.

Found with TorchFix https://github.com/pytorch-labs/torchfix/

* Use non-alias
2025-01-19 04:54:32 -05:00
Pam
21ac610254 Rewrite res_multistep sampler and implement res_multistep_cfg_pp sampler. (#6462) 2025-01-14 18:20:06 -05:00
comfyanonymous
302635bf4d Add res_multistep sampler from the cosmos code.
This sampler should work with all models.
2025-01-12 03:10:07 -05:00
comfyanonymous
c75fa1bbf4 Make a few more samplers deterministic. 2025-01-07 02:12:32 -05:00
comfyanonymous
11fe0e25f3 Make ancestral samplers more deterministic. 2025-01-06 03:04:32 -05:00
comfyanonymous
5ff702f0bf Remove some trailing white space. 2024-12-27 18:02:30 -05:00
Chenlei Hu
efca88cfb3 Lint all unused variables (#5989)
* Enable F841

* Autofix

* Remove all unused variable assignment
2024-12-12 17:59:16 -05:00
comfyanonymous
0ebecf6d11 Some optimizations to euler a. 2024-12-03 06:11:52 -05:00
comfyanonymous
c690e318b1 Remove useless code. 2024-12-02 06:49:55 -05:00
comfyanonymous
797c883f33 make dpm_2_ancestral work with rectified flow. 2024-12-01 07:37:41 -05:00
comfyanonymous
d3c29b6c15 Make euler_ancestral work on flow models (credit: Ashen). 2024-10-25 19:53:44 -04:00
comfyanonymous
3eeaf6f13f Cleanup some useless lines. 2024-10-14 21:02:39 -04:00
chaObserv
b3ca8146a0 Add dpmpp_2m_cfg_pp (#4992) 2024-09-24 02:42:56 -04:00
yoinked
65b99536ee add laplace scheduler [2407.03297] (#4990)
* add laplace scheduler [2407.03297]

* should be here instead lol

* better settings
2024-09-19 23:23:09 -04:00
comfyanonymous
2e36788c9c Doesn't really make sense for cfg_pp sampler to call regular one. 2024-09-11 02:51:36 -04:00
yoinked
f482299948 Add CFG++ to DPM++ 2S Ancestral (#3871)
* Update sampling.py

* Update samplers.py

* my bad

* "fix" the sampler

* Update samplers.py

* i named it wrong

* minor sampling improvements

mainly using a dynamic rho value (hey this sounds a lot like smea!!!)

* revert rho change

rho? r? its just 1/2
2024-09-11 02:49:44 -04:00
comfyanonymous
6d23fd0618 Automatically use RF variant of dpmpp_2s_ancestral if RF model. 2024-08-18 00:47:25 -04:00
Ashen
41b3ccc486 dpmpp_2s_ancestral_RF for rectified flow (Flux, SD3 and Auraflow). 2024-08-18 00:33:30 -04:00
comfyanonymous
a7050b3f3b Switch to the real cfg++ method in the samplers.
The old _pp ones will be updated automatically to the regular ones with 2x
the cfg.

My fault for not checking what the "_pp" samplers actually did.
2024-06-29 11:59:48 -04:00
comfyanonymous
ad1d7cb5c2 Remove print. 2024-06-27 02:54:15 -04:00
comfyanonymous
561a0d0c62 Add DEIS order 3 sampler.
Order 4 seems to give bad results.
2024-06-26 22:40:05 -04:00
comfyanonymous
8940c4800b Implement my alternative take on CFG++ as the euler_pp sampler.
Add euler_ancestral_pp which is the ancestral version of euler with the
same modification.
2024-06-25 07:41:52 -04:00
comfyanonymous
0bae9a67e5 Add ipndm_v sampler, works best with the exponential scheduler. 2024-06-20 08:51:49 -04:00
comfyanonymous
9c26d62191 Add ipndm sampler. 2024-06-19 16:32:30 -04:00
comfyanonymous
5c5911a0c1 Optimize some unneeded if conditions in the sampling code. 2024-06-15 02:26:19 -04:00
comfyanonymous
b04267ce30 Fix some edge cases with samplers and arrays with a single sigma. 2024-05-01 17:05:30 -04:00
comfyanonymous
4e8d2577a9 Fix regression with lcm not working with batches. 2024-03-20 20:48:54 -04:00
comfyanonymous
90ef1527e8 Make LCM sampler use the model noise scaling function. 2024-03-20 01:35:59 -04:00
comfyanonymous
e5eeb33ac4 heunpp2 sampler. 2023-11-14 23:50:55 -05:00
comfyanonymous
43dcfcd754 Support lcm models.
Use the "lcm" sampler to sample them, you also have to use the
ModelSamplingDiscrete node to set them as lcm models to use them properly.
2023-11-09 18:30:22 -05:00
comfyanonymous
ff096742bd Not used anymore. 2023-11-01 00:01:30 -04:00
comfyanonymous
370c837794 Fix uni_pc sampler math. This changes the images this sampler produces. 2023-10-20 04:16:53 -04:00
comfyanonymous
509dd5ca26 Add DDPM sampler. 2023-09-15 19:22:47 -04:00
Alexopus
a5a8d25943 Fix referenced before assignment
For https://github.com/BlenderNeko/ComfyUI_TiledKSampler/issues/13
2023-08-17 22:30:07 +02:00
comfyanonymous
87f1037cf5 Remove 3m from PR #1213 because of some small issues. 2023-08-14 00:48:45 -04:00
comfyanonymous
c5666c503b Gpu variant of dpmpp_3m_sde. Note: use 3m with exponential or karras. 2023-08-14 00:28:50 -04:00
FizzleDorf
8b4773ee86 dpmpp 3m + dpmpp 3m sde added 2023-08-13 22:29:04 -04:00