CleanDiffuserTeam / CleanDiffuser

CleanDiffuser: An Easy-to-use Modularized Library for Diffusion Models in Decision Making
Apache License 2.0
184 stars 16 forks source link

add Consistency Models #7

Closed ZibinDong closed 1 week ago

ZibinDong commented 1 week ago

[2024-06-24] 🥰 We have added Consistency Models into CleanDiffuser. With one model, you can do both Consistency Distillation and Consistency Training! Check out an example in tutorials/sp_consistency_policy.py! (Note: Our consistency training implementation refers to the improved version, see https://arxiv.org/abs/2310.14189.)