mirror of
https://github.com/huggingface/diffusers.git
synced 2026-10-26 08:59:24 +03:00
351de3a6fe
* docs
* update
* Revert "docs"
This reverts commit 04ca80543f.
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
1.2 KiB
1.2 KiB
CosmosTransformer3DModel
A Diffusion Transformer model for 3D video-like data was introduced in Cosmos World Foundation Model Platform for Physical AI by NVIDIA.
The model can be loaded with the following code snippet.
from diffusers import CosmosTransformer3DModel
transformer = CosmosTransformer3DModel.from_pretrained("nvidia/Cosmos-1.0-Diffusion-7B-Text2World", subfolder="transformer", dtype=torch.bfloat16)
CosmosTransformer3DModel
autodoc CosmosTransformer3DModel
Transformer2DModelOutput
autodoc models.modeling_outputs.Transformer2DModelOutput