A Diffusion Transformer model for 3D video-like data was introduced in Wan Animate by the Alibaba Wan Team.
The model can be loaded with the following code snippet.
from diffusers import WanAnimateTransformer3DModel
transformer = WanAnimateTransformer3DModel.from_pretrained("Wan-AI/Wan2.2-Animate-14B-Diffusers", subfolder="transformer", torch_dtype=torch.bfloat16)
[[autodoc]] WanAnimateTransformer3DModel
[[autodoc]] models.modeling_outputs.Transformer2DModelOutput