AllegroTransformer3DModel

Allegro 的 3D 数据扩散 Transformer 模型在 RhymesAI 的论文 Allegro: Open the Black Box of Commercial-Level Video Generation Model 中被介绍。

可以使用以下代码片段加载该模型。

from diffusers import AllegroTransformer3DModel

vae = AllegroTransformer3DModel.from_pretrained("rhymes-ai/Allegro", subfolder="transformer", torch_dtype=torch.bfloat16).to("cuda")

AllegroTransformer3DModel

[[autodoc]] AllegroTransformer3DModel

Transformer2DModelOutput

[[autodoc]] models.modeling_outputs.Transformer2DModelOutput