Unverified Commit 04e783cd authored by Aryan's avatar Aryan Committed by GitHub
Browse files

Update variable names correctly in docs (#10435)

fix
parent 1896b1f7
...@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet. ...@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet.
```python ```python
from diffusers import AllegroTransformer3DModel from diffusers import AllegroTransformer3DModel
vae = AllegroTransformer3DModel.from_pretrained("rhymes-ai/Allegro", subfolder="transformer", torch_dtype=torch.bfloat16).to("cuda") transformer = AllegroTransformer3DModel.from_pretrained("rhymes-ai/Allegro", subfolder="transformer", torch_dtype=torch.bfloat16).to("cuda")
``` ```
## AllegroTransformer3DModel ## AllegroTransformer3DModel
......
...@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet. ...@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet.
```python ```python
from diffusers import CogVideoXTransformer3DModel from diffusers import CogVideoXTransformer3DModel
vae = CogVideoXTransformer3DModel.from_pretrained("THUDM/CogVideoX-2b", subfolder="transformer", torch_dtype=torch.float16).to("cuda") transformer = CogVideoXTransformer3DModel.from_pretrained("THUDM/CogVideoX-2b", subfolder="transformer", torch_dtype=torch.float16).to("cuda")
``` ```
## CogVideoXTransformer3DModel ## CogVideoXTransformer3DModel
......
...@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet. ...@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet.
```python ```python
from diffusers import CogView3PlusTransformer2DModel from diffusers import CogView3PlusTransformer2DModel
vae = CogView3PlusTransformer2DModel.from_pretrained("THUDM/CogView3Plus-3b", subfolder="transformer", torch_dtype=torch.bfloat16).to("cuda") transformer = CogView3PlusTransformer2DModel.from_pretrained("THUDM/CogView3Plus-3b", subfolder="transformer", torch_dtype=torch.bfloat16).to("cuda")
``` ```
## CogView3PlusTransformer2DModel ## CogView3PlusTransformer2DModel
......
...@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet. ...@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet.
```python ```python
from diffusers import MochiTransformer3DModel from diffusers import MochiTransformer3DModel
vae = MochiTransformer3DModel.from_pretrained("genmo/mochi-1-preview", subfolder="transformer", torch_dtype=torch.float16).to("cuda") transformer = MochiTransformer3DModel.from_pretrained("genmo/mochi-1-preview", subfolder="transformer", torch_dtype=torch.float16).to("cuda")
``` ```
## MochiTransformer3DModel ## MochiTransformer3DModel
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment