diffusers

Commit Graph

Author	SHA1	Message	Date
Patrick von Platen	88fa6b7d68	[Dance Diffusion] Add dance diffusion (#803 ) * start * add more logic * Update src/diffusers/models/unet_2d_condition_flax.py * match weights * up * make model work * making class more general, fixing missed file rename * small fix * make new conversion work * up * finalize conversion * up * first batch of variable renamings * remove c and c_prev var names * add mid and out block structure * add pipeline * up * finish conversion * finish * upload * more fixes * Apply suggestions from code review * add attr * up * uP * up * finish tests * finish * uP * finish * fix test * up * naming consistency in tests * Apply suggestions from code review Co-authored-by: Suraj Patil <surajp815@gmail.com> Co-authored-by: Pedro Cuenca <pedro@huggingface.co> Co-authored-by: Nathan Lambert <nathan@huggingface.co> Co-authored-by: Anton Lozhkov <anton@huggingface.co> * remove hardcoded 16 * Remove bogus * fix some stuff * finish * improve logging * docs * upload Co-authored-by: Nathan Lambert <nol@berkeley.edu> Co-authored-by: Suraj Patil <surajp815@gmail.com> Co-authored-by: Pedro Cuenca <pedro@huggingface.co> Co-authored-by: Nathan Lambert <nathan@huggingface.co> Co-authored-by: Anton Lozhkov <anton@huggingface.co>	2022-10-25 18:39:25 +02:00
anton-l	970e30606c	Revert "[v0.4.0] Temporarily remove Flax modules from the public API (#755 )" This reverts commit `2e209c30cf`.	2022-10-06 18:35:40 +02:00
Anton Lozhkov	2e209c30cf	[v0.4.0] Temporarily remove Flax modules from the public API (#755 ) Temporarily remove Flax modules from the public API	2022-10-06 18:10:36 +02:00
Younes Belkada	8b0be93596	Flax documentation (#589 ) * documenting `attention_flax.py` file * documenting `embeddings_flax.py` * documenting `unet_blocks_flax.py` * Add new objs to doc page * document `vae_flax.py` * Apply suggestions from code review * modify `unet_2d_condition_flax.py` * make style * Apply suggestions from code review * make style * Apply suggestions from code review * fix indent * fix typo * fix indent unet * Update src/diffusers/models/vae_flax.py * Apply suggestions from code review Co-authored-by: Pedro Cuenca <pedro@huggingface.co> Co-authored-by: Mishig Davaadorj <dmishig@gmail.com> Co-authored-by: Pedro Cuenca <pedro@huggingface.co>	2022-09-23 13:24:16 +02:00
Kashif Rasul	5e6417e988	[Docs] Models (#416 ) * docs for attention * types for embeddings * unet2d docstrings * UNet2DConditionModel docstrings * fix typos * style and vq-vae docstrings * docstrings for VAE * Update src/diffusers/models/unet_2d.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * make style * added inherits from sentence * docstring to forward * make style * Apply suggestions from code review Co-authored-by: Pedro Cuenca <pedro@huggingface.co> * finish model docs * up Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> Co-authored-by: Pedro Cuenca <pedro@huggingface.co>	2022-09-08 17:28:11 +02:00
Patrick von Platen	5a38033de4	[Docs] Let's go (#385 )	2022-09-07 11:31:13 +02:00

6 Commits