Skip to content

Commit ad5ecd1

Browse files
authored
[docs] Fix CogVideoX table (huggingface#10008)
* fix * fix
1 parent 074e123 commit ad5ecd1

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

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

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,15 +30,17 @@ Make sure to check out the Schedulers [guide](../../using-diffusers/schedulers.m
3030
This pipeline was contributed by [zRzRzRzRzRzRzR](https://github.com/zRzRzRzRzRzRzR). The original codebase can be found [here](https://huggingface.co/THUDM). The original weights can be found under [hf.co/THUDM](https://huggingface.co/THUDM).
3131

3232
There are three official CogVideoX checkpoints for text-to-video and video-to-video.
33+
3334
| checkpoints | recommended inference dtype |
34-
|---|---|
35+
|:---:|:---:|
3536
| [`THUDM/CogVideoX-2b`](https://huggingface.co/THUDM/CogVideoX-2b) | torch.float16 |
3637
| [`THUDM/CogVideoX-5b`](https://huggingface.co/THUDM/CogVideoX-5b) | torch.bfloat16 |
3738
| [`THUDM/CogVideoX1.5-5b`](https://huggingface.co/THUDM/CogVideoX1.5-5b) | torch.bfloat16 |
3839

3940
There are two official CogVideoX checkpoints available for image-to-video.
41+
4042
| checkpoints | recommended inference dtype |
41-
|---|---|
43+
|:---:|:---:|
4244
| [`THUDM/CogVideoX-5b-I2V`](https://huggingface.co/THUDM/CogVideoX-5b-I2V) | torch.bfloat16 |
4345
| [`THUDM/CogVideoX-1.5-5b-I2V`](https://huggingface.co/THUDM/CogVideoX-1.5-5b-I2V) | torch.bfloat16 |
4446

@@ -48,8 +50,9 @@ For the CogVideoX 1.5 series:
4850
- Both T2V and I2V models support generation with 81 and 161 frames and work best at this value. Exporting videos at 16 FPS is recommended.
4951

5052
There are two official CogVideoX checkpoints that support pose controllable generation (by the [Alibaba-PAI](https://huggingface.co/alibaba-pai) team).
53+
5154
| checkpoints | recommended inference dtype |
52-
|---|---|
55+
|:---:|:---:|
5356
| [`alibaba-pai/CogVideoX-Fun-V1.1-2b-Pose`](https://huggingface.co/alibaba-pai/CogVideoX-Fun-V1.1-2b-Pose) | torch.bfloat16 |
5457
| [`alibaba-pai/CogVideoX-Fun-V1.1-5b-Pose`](https://huggingface.co/alibaba-pai/CogVideoX-Fun-V1.1-5b-Pose) | torch.bfloat16 |
5558

0 commit comments

Comments
 (0)