Pierre Chapuis
ed7e2e5964
check timesteps stay the same in bfloat16
2024-09-26 08:55:07 +02:00
Pierre Chapuis
f1c8503e68
use default (1e-8) atol for basic DPM solver (used in SD1.5)
2024-09-26 08:55:07 +02:00
Pierre Chapuis
83b931296f
fix DPM-Solver with bfloat16
CI / lint_and_typecheck (push) Waiting to run
Deploy docs to GitHub Pages / Deploy docs (push) Waiting to run
Spell checker / Spell check (push) Waiting to run
2024-09-25 14:41:19 +02:00
limiteinductive
af6c5aecbe
add karras sigmas to dpm solver
2024-09-06 15:30:02 +02:00
limiteinductive
09a9dfd494
Add stochastic sampling to DPM solver (SDE)
CI / lint_and_typecheck (push) Waiting to run
Deploy docs to GitHub Pages / Deploy docs (push) Waiting to run
Spell checker / Spell check (push) Waiting to run
2024-07-23 11:13:12 +02:00
Pierre Chapuis
daee77298d
improve FrankenSolver
...
It now takes a Scheduler factory instead of a Scheduler.
This lets the user potentially recreate the Scheduler on `rebuild`.
It also properly sets the device and dtype on rebuild,
and it has better typing.
2024-07-19 16:46:52 +02:00
Pierre Chapuis
b4db08de24
use "solver" (not scheduler) wording in tests
2024-07-10 19:31:34 +02:00
Pierre Chapuis
9e8c2a3753
add FrankenSolver
...
This solver is designed to use Diffusers Schedulers as Refiners Solvers.
2024-07-10 19:31:34 +02:00
Laurent
17246708b9
Add sample_noise
staticmethod and modify add_noise
to support batched steps
2024-04-18 12:55:49 +02:00
Pierre Chapuis
bf0ba58541
refactor solver params, add sample prediction type
2024-02-26 12:14:02 +01:00
Pierre Chapuis
ddc1cf8ca7
refactor solvers to support different timestep spacings
2024-02-26 12:14:02 +01:00
Pierre Chapuis
c8c6294550
add LCMSolver (Latent Consistency Models)
2024-02-21 16:37:27 +01:00
Pierre Chapuis
3de1508b65
increase tolerance on Euler test
2024-02-04 08:58:22 +01:00
Pierre Chapuis
83b478c0ff
fix test failure caused by Diffusers 0.26.0
2024-02-04 08:58:22 +01:00
Pierre Chapuis
3ddd258d36
add a test for noise schedules
2024-02-01 16:17:07 +01:00
limiteinductive
73f6ccfc98
make Scheduler a fl.Module + Change name Scheduler -> Solver
2024-01-31 17:03:52 +01:00