models

Models

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

Name	Name	Last commit message	Last commit date
parent directory ..
autoencoders	autoencoders
transformers	transformers
unets	unets
README.md	README.md
__init__.py	__init__.py
activations.py	activations.py
adapter.py	adapter.py
attention.py	attention.py
attention_flax.py	attention_flax.py
attention_processor.py	attention_processor.py
controlnet.py	controlnet.py
controlnet_flax.py	controlnet_flax.py
downsampling.py	downsampling.py
dual_transformer_2d.py	dual_transformer_2d.py
embeddings.py	embeddings.py
embeddings_flax.py	embeddings_flax.py
lora.py	lora.py
modeling_flax_pytorch_utils.py	modeling_flax_pytorch_utils.py
modeling_flax_utils.py	modeling_flax_utils.py
modeling_outputs.py	modeling_outputs.py
modeling_pytorch_flax_utils.py	modeling_pytorch_flax_utils.py
modeling_utils.py	modeling_utils.py
normalization.py	normalization.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_condition.py	unet_2d_condition.py
upsampling.py	upsampling.py
vae_flax.py	vae_flax.py
vq_model.py	vq_model.py