Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
04e783cd
Unverified
Commit
04e783cd
authored
Jan 06, 2025
by
Aryan
Committed by
GitHub
Jan 06, 2025
Browse files
Update variable names correctly in docs (#10435)
fix
parent
1896b1f7
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
4 additions
and
4 deletions
+4
-4
docs/source/en/api/models/allegro_transformer3d.md
docs/source/en/api/models/allegro_transformer3d.md
+1
-1
docs/source/en/api/models/cogvideox_transformer3d.md
docs/source/en/api/models/cogvideox_transformer3d.md
+1
-1
docs/source/en/api/models/cogview3plus_transformer2d.md
docs/source/en/api/models/cogview3plus_transformer2d.md
+1
-1
docs/source/en/api/models/mochi_transformer3d.md
docs/source/en/api/models/mochi_transformer3d.md
+1
-1
No files found.
docs/source/en/api/models/allegro_transformer3d.md
View file @
04e783cd
...
@@ -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
...
...
docs/source/en/api/models/cogvideox_transformer3d.md
View file @
04e783cd
...
@@ -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
...
...
docs/source/en/api/models/cogview3plus_transformer2d.md
View file @
04e783cd
...
@@ -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
...
...
docs/source/en/api/models/mochi_transformer3d.md
View file @
04e783cd
...
@@ -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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment