File tree Expand file tree Collapse file tree 2 files changed +32
-0
lines changed Expand file tree Collapse file tree 2 files changed +32
-0
lines changed Original file line number Diff line number Diff line change 266266 title : LatteTransformer3DModel
267267 - local : api/models/lumina_nextdit2d
268268 title : LuminaNextDiT2DModel
269+ - local : api/models/mochi_transformer3d
270+ title : MochiTransformer3DModel
269271 - local : api/models/pixart_transformer2d
270272 title : PixArtTransformer2DModel
271273 - local : api/models/prior_transformer
Original file line number Diff line number Diff line change 1+ <!-- Copyright 2024 The HuggingFace Team. All rights reserved.
2+
3+ Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
4+ the License. You may obtain a copy of the License at
5+
6+ http://www.apache.org/licenses/LICENSE-2.0
7+
8+ Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
9+ an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
10+ specific language governing permissions and limitations under the License. -->
11+
12+ # MochiTransformer3DModel
13+
14+ A Diffusion Transformer model for 3D video-like data was introduced in [ Mochi-1 Preview] ( https://huggingface.co/genmo/mochi-1-preview ) by Genmo.
15+
16+ The model can be loaded with the following code snippet.
17+
18+ ``` python
19+ from diffusers import MochiTransformer3DModel
20+
21+ vae = MochiTransformer3DModel.from_pretrained(" genmo/mochi-1-preview" , subfolder = " transformer" , torch_dtype = torch.float16).to(" cuda" )
22+ ```
23+
24+ ## MochiTransformer3DModel
25+
26+ [[ autodoc]] MochiTransformer3DModel
27+
28+ ## Transformer2DModelOutput
29+
30+ [[ autodoc]] models.modeling_outputs.Transformer2DModelOutput
You can’t perform that action at this time.
0 commit comments