models

Models

For more detail on the models, please refer to the docs.

Name		Name	Last commit message	Last commit date
parent directory ..
README.md		README.md
__init__.py		__init__.py
activations.py		activations.py
attention.py		attention.py
attention_flax.py		attention_flax.py
attention_processor.py		attention_processor.py
autoencoder_kl.py		autoencoder_kl.py
controlnet.py		controlnet.py
controlnet_flax.py		controlnet_flax.py
cross_attention.py		cross_attention.py
dual_transformer_2d.py		dual_transformer_2d.py
embeddings.py		embeddings.py
embeddings_flax.py		embeddings_flax.py
modeling_flax_pytorch_utils.py		modeling_flax_pytorch_utils.py
modeling_flax_utils.py		modeling_flax_utils.py
modeling_pytorch_flax_utils.py		modeling_pytorch_flax_utils.py
modeling_utils.py		modeling_utils.py
prior_transformer.py		prior_transformer.py
resnet.py		resnet.py
resnet_flax.py		resnet_flax.py
t5_film_transformer.py		t5_film_transformer.py
transformer_2d.py		transformer_2d.py
transformer_temporal.py		transformer_temporal.py
unet_1d.py		unet_1d.py
unet_1d_blocks.py		unet_1d_blocks.py
unet_2d.py		unet_2d.py
unet_2d_blocks.py		unet_2d_blocks.py
unet_2d_blocks_flax.py		unet_2d_blocks_flax.py
unet_2d_condition.py		unet_2d_condition.py
unet_2d_condition_flax.py		unet_2d_condition_flax.py
unet_3d_blocks.py		unet_3d_blocks.py
unet_3d_condition.py		unet_3d_condition.py
vae.py		vae.py
vae_flax.py		vae_flax.py
vq_model.py		vq_model.py