diff options
| author | Nick Senger <dev@nsenger.com> | 2025-01-01 12:34:17 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-01-01 21:34:17 +0100 |
| commit | cbaa0ad46f0eda2f3d9bcf8a42d6271e6760e578 (patch) | |
| tree | f8a899e47e907bcf79f877e0cb307d2b5d57db1f /candle-transformers/src/models/stable_diffusion/mod.rs | |
| parent | b12c7c2888c49e7f133bb2dc29f8fdbb04a37e10 (diff) | |
| download | candle-cbaa0ad46f0eda2f3d9bcf8a42d6271e6760e578.tar.gz candle-cbaa0ad46f0eda2f3d9bcf8a42d6271e6760e578.tar.bz2 candle-cbaa0ad46f0eda2f3d9bcf8a42d6271e6760e578.zip | |
UniPC for diffusion sampling (#2684)
* feat: Add unipc multistep scheduler
* chore: Clippy and formatting
* chore: Update comments
* chore: Avoid unsafety in float ordering
* refactor: Update Scheduler::step mutability requirements
* fix: Corrector img2img
* chore: Update unipc ref link to latest diffusers release
* chore: Deduplicate float ordering
* fix: Panic when running with dev profile
Diffstat (limited to 'candle-transformers/src/models/stable_diffusion/mod.rs')
| -rw-r--r-- | candle-transformers/src/models/stable_diffusion/mod.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-transformers/src/models/stable_diffusion/mod.rs b/candle-transformers/src/models/stable_diffusion/mod.rs index 6d89f9cd..4c685209 100644 --- a/candle-transformers/src/models/stable_diffusion/mod.rs +++ b/candle-transformers/src/models/stable_diffusion/mod.rs @@ -47,6 +47,7 @@ pub mod resnet; pub mod schedulers; pub mod unet_2d; pub mod unet_2d_blocks; +pub mod uni_pc; pub mod utils; pub mod vae; |