Skip to content

Commit e063e9d

Browse files
a-r-r-o-wstevhliu
andauthored
Update docs/source/en/api/pipelines/cogvideox.md
Co-authored-by: Steven Liu <[email protected]>
1 parent 2c3b78d commit e063e9d

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

docs/source/en/api/pipelines/cogvideox.md

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,11 @@ There are three official CogVideoX checkpoints for text-to-video and video-to-vi
3636
| [`THUDM/CogVideoX-5b`](https://huggingface.co/THUDM/CogVideoX-5b) | torch.bfloat16 |
3737
| [`THUDM/CogVideoX-1.5-5b`](https://huggingface.co/THUDM/CogVideoX-1.5-5b) | torch.bfloat16 |
3838

39-
There is one model available that can be used with the image-to-video CogVideoX pipeline:
39+
There are two official CogVideoX checkpoints available for image-to-video.
40+
| checkpoints | recommended inference dtype |
41+
|---|---|
42+
| [`THUDM/CogVideoX-5b-I2V`](https://huggingface.co/THUDM/CogVideoX-5b-I2V) | torch.bfloat16 |
43+
| [`THUDM/CogVideoX-1.5-5b-I2V`](https://huggingface.co/THUDM/CogVideoX-1.5-5b-I2V) | torch.bfloat16 |
4044
- [`THUDM/CogVideoX-5b-I2V`](https://huggingface.co/THUDM/CogVideoX-5b-I2V): The recommended dtype for running this model is `torch.bfloat16`.
4145
- [`THUDM/CogVideoX-1.5-5b-I2V`](https://huggingface.co/THUDM/CogVideoX-1.5-5b-I2V): The recommended dtype for running this mdoel is `torch.bfloat16`.
4246

0 commit comments

Comments
 (0)