3 Commits

Author	SHA1	Message	Date
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