diffusers
Add Consistency Models Pipeline
#3492
Merged
Overview
Commits
79
Changes
View On GitHub

Add Consistency Models Pipeline #3492

patrickvonplaten merged 79 commits into huggingface:main from dg845:consistency-models-pipeline
dg845
dg845 initial commit
799ab237
dg845 Improve consistency models sampling implementation.
63b7f01b
dg845 Add CMStochasticIterativeScheduler, which implements the multi-step s…
f2e53da3
dg845 dg845 changed the title Consistency models pipeline [WIP] Add Consistency models pipeline 2 years ago
dg845 dg845 changed the title [WIP] Add Consistency models pipeline [WIP] Add Consistency Models Pipeline 2 years ago
HuggingFaceDocBuilderDev
dg845
ayushtues
ayushtues
ayushtues
dg845
dg845
ayushtues
Merge branch 'main2' into consistency_unet
37e1895b
Add Unet blocks for consistency models
7c1e81f8
ayushtues
Add conversion script for Unet
3a151bd1
Fix bug in new unet blocks
b6c5e15b
patrickvonplaten
williamberman
dg845
Fix attention weight loading
4e93f09f
ayushtues
patrickvonplaten
williamberman
dg845
williamberman
dg845 Merge pull request #1 from ayushtues/consistency_unet
d137d11b
ayushtues
williamberman
dg845 Make design improvements to ConsistencyModelPipeline and CMStochastic…
9ae7669f
dg845 make style
54b287e7
ayushtues
ayushtues
williamberman
dg845 Make small random test UNet class conditional and set resnet_time_sca…
5a27c2f4
dg845 Add support for converting a test UNet and non-class-conditional UNet…
f2783a87
dg845 make style
ed53b859
dg845
dg845 Change num_class_embeds to 1000 to better match the original consiste…
a505c6cc
dg845 Add support for distillation in pipeline_consistency_models.py.
a927a4af
dg845 Improve consistency model tests:
b2e64243
dg845 make style
c37e302d
dg845 Improve ConsistencyModelPipeline:
a0a164cf
dg845 make style
0d1de089
dg845 Improve ConsistencyModelPipeline:
5f4f4064
dg845 make style
213b25de
dg845 Fix bug when randomly generating class labels for class-conditional g…
fbe34c3f
dg845 Switch CMStochasticIterativeScheduler to configuring a sigma schedule…
0e53d8bb
dg845 Remove some unused code and make style.
c3b242ed
dg845 Fix small bug in CMStochasticIterativeScheduler.
43e43792
dg845 Add expected slices for multistep sampling tests and make them pass.
94c99ca4
dg845 Work on consistency model fast tests:
0773b271
dg845 make style
6867a3ad
patrickvonplaten
dg845 Refactor conversion script to make it easier to add more model archit…
f0c85d3e
dg845 Work on ConsistencyModelPipeline tests:
6adb5892
dg845 make style
c213bf7e
dg845 Remove the add_noise and add_noise_to_input methods from CMStochastic…
3be6e086
dg845 Run python utils/check_copies.py --fix_and_overwrite
660909f8
dg845 Make fast tests from PipelineTesterMixin pass.
ad5abdc0
dg845 make style
a1c11d30
dg845
ayushtues
patrickvonplaten
williamberman
williamberman
williamberman commented on 2023-06-05
williamberman
dg845
dg845 Refactor consistency models pipeline and scheduler:
2dc2fa31
dg845 make style
7d3dbe3d
dg845
dg845 Add docstrings and further refactor pipeline and scheduler.
55f80ed9
dg845 make style
3234543f
dg845
dg845
dg845 Add initial version of the consistency models documentation.
dc4349ca
williamberman
williamberman
williamberman
williamberman commented on 2023-06-09
williamberman
williamberman commented on 2023-06-09
williamberman
williamberman commented on 2023-06-09
ayushtues
ayushtues
dg845
dg845 Refactor custom timesteps logic following DDPMScheduler/IFPipeline an…
85eb7961
dg845 make style
bf3a4056
ayushtues
ayushtues
patrickvonplaten
dg845
dg845 Convert current slow tests to use fp16 and flash attention.
fd0a2535
dg845 make style
7c03ff6d
dg845
dg845 Add slow tests for normal attention on cuda device.
80d77452
dg845 make style
0662f63f
ayushtues Fix attention weights loading
cce04c09
ayushtues
dg845 Merge pull request #3 from ayushtues/fix-attention-loading
c1ba8b92
dg845 Update consistency model fast tests for new test checkpoints with att…
62a49a2f
dg845 make style
58f12cad
dg845
williamberman
williamberman commented on 2023-06-15
williamberman
williamberman commented on 2023-06-15
williamberman
williamberman commented on 2023-06-15
williamberman
williamberman commented on 2023-06-15
williamberman
williamberman commented on 2023-06-15
williamberman
williamberman commented on 2023-06-15
williamberman
williamberman commented on 2023-06-15
williamberman
dg845 apply suggestions
071f8506
dg845 Add add_noise method to CMStochasticIterativeScheduler (copied from E…
69686154
dg845 Conversion script now outputs pipeline instead of UNet and add suppor…
ca15734d
dg845 When both timesteps and num_inference_steps are supplied, raise warni…
a56d3d28
dg845 make style
a17847ee
dg845 Add remaining diffusers model checkpoints for models in the original …
8214a33e
dg845
dg845
patrickvonplaten
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
patrickvonplaten commented on 2023-06-28
patrickvonplaten
dg845 apply suggestions from review
2606d82e
dg845 make style
075351ad
dg845
patrickvonplaten
patrickvonplaten commented on 2023-06-29
patrickvonplaten
dg845 Merge branch 'main' into consistency-models-pipeline
89764711
dg845
dg845 fix attention naming
64860490
dg845
dg845 Add tests for CMStochasticIterativeScheduler.
cf514b37
dg845 make style
b5c2b233
dg845
patrickvonplaten
dg845
dg845 Make CMStochasticIterativeScheduler tests pass.
60b05271
dg845 make style
7670cd8b
dg845
dg845 Override test_step_shape in CMStochasticIterativeSchedulerTest instea…
cb48162c
dg845 make style
ee9f8bfc
sayakpaul
sayakpaul commented on 2023-07-05
sayakpaul
sayakpaul commented on 2023-07-05
sayakpaul
sayakpaul commented on 2023-07-05
sayakpaul
patrickvonplaten rename some models
2fc7f2c5
patrickvonplaten Improve API
e14189c5
patrickvonplaten rename some models
9f627ed9
patrickvonplaten Remove duplicated block
3febb18f
patrickvonplaten Add docstring and make torch compile work
d430fab8
patrickvonplaten Merge branch 'main' into consistency-models-pipeline
44d2c223
patrickvonplaten
patrickvonplaten
patrickvonplaten More fixes
f7675d39
patrickvonplaten Fixes
c66b7022
patrickvonplaten patrickvonplaten changed the title [WIP] Add Consistency Models Pipeline Add Consistency Models Pipeline 2 years ago
patrickvonplaten
patrickvonplaten commented on 2023-07-05
patrickvonplaten
patrickvonplaten commented on 2023-07-05
patrickvonplaten Apply suggestions from code review
66f1603b
patrickvonplaten
patrickvonplaten commented on 2023-07-05
patrickvonplaten
patrickvonplaten commented on 2023-07-05
patrickvonplaten Apply suggestions from code review
033fbe67
patrickvonplaten add more docstring
fd67410b
patrickvonplaten
patrickvonplaten commented on 2023-07-05
patrickvonplaten
patrickvonplaten approved these changes on 2023-07-05
dg845
patrickvonplaten update consistency conversion script
541dd842
williamberman
williamberman commented on 2023-07-05
williamberman
williamberman approved these changes on 2023-07-05
patrickvonplaten patrickvonplaten merged aed7499a into main 2 years ago
svjack

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum