Add Latent Consistency Models Pipeline #5448
initial commit for LatentConsistencyModelPipeline and LCMScheduler ba…
904b7f57
Add callback and freeu support.
4e25452c
apply suggestions from review
3eec9802
Clean up LCMScheduler
cadb86bd
Remove timeindex argument to LCMScheduler.step.
0740f367
Add support for clipping or thresholding the predicted original sample.
b9600377
Remove unused methods and arguments in LCMScheduler.
468511a0
Improve comment about (lack of) negative prompt support.
6b007567
Change input guidance_scale to match the StableDiffusionPipeline (Ima…
cbc4d557
Move lcm_origin_steps from pipeline __call__ to LCMScheduler.__init__…
e8f65bfc
a-r-r-o-w
approved these changes
on 2023-10-20
Fix typo when clipping/thresholding in LCMScheduler.
f8005478
Add some initial LCMScheduler tests.
21cb605b
add type annotations from review
ae358aca
Fix type annotation bug.
00f4b48e
Override test_add_noise_device in LCMSchedulerTest since hardcoded ti…
959e7f8d
Add generator argument pipeline prepare_latents call.
696d9877
Cast LCMScheduler.timesteps to long in set_timesteps.
43361b7e
Add onestep and multistep full loop scheduler tests.
13a54b8f
Set default height/width to None and don't hardcode guidance scale em…
a61b4aef
Add initial LatentConsistencyPipeline fast and slow tests.
3de14655
Add initial documentation for LatentConsistencyModelPipeline and LCMS…
cd4b0ab6
dg845
marked this pull request as ready for review 2 years ago
Merge branch 'main' into latent-consistency-models-pipeline
a3216395
Make remaining failing fast tests pass.
9df89eac
make style
0dd2a829
Make original_inference_steps configurable from pipeline __call__ again.
cbbd1874
make style
8868d349
Remove guidance_rescale arg from pipeline __call__ since LCM currentl…
92b68b27
Make LCMScheduler defaults match config of LCM_Dreamshaper_v7 checkpo…
24c8cad6
Fix LatentConsistencyPipeline slow tests and add dummy expected slices.
e191555d
Add checks for original_steps in LCMScheduler.set_timesteps.
e439118b
make fix-copies
516d8fd7
Improve LatentConsistencyModelPipeline docs.
e57913d1
dg845
changed the title [WIP] Add Latent Consistency Models Pipeline Add Latent Consistency Models Pipeline 2 years ago
Apply suggestions from code review
6faa29f4
Apply suggestions from code review
b08df598
Apply suggestions from code review
264bd77e
Update src/diffusers/schedulers/scheduling_lcm.py
888b96d8
Apply suggestions from code review
686c0c06
Merge branch 'main' into latent-consistency-models-pipeline
2cd10195
finish
2f2d14c5
dg845
deleted the latent-consistency-models-pipeline branch 2 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub