Instructions to use zai-org/CogVideoX-2b with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use zai-org/CogVideoX-2b with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("zai-org/CogVideoX-2b", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Commit History
-
Update vae config to remove unused parameters
Verified signatureHey! These parameters are no longer supported in the Diffusers VAE implementation. They weren't used to begin with, but made its way into the codebase after reviews. They've now been remove which causes the following warning is raised when running inference with CogVideoX-2b: ``` The config attributes {'mid_block_add_attention': True, 'sample_size': 256} were passed to AutoencoderKLCogVideoX, but are not expected and will be ignored. Please verify your config.json configuration file. ``` This PR will get rid of the warning. Thanks! cc @zRzRzRzRzRzRzR102080d -
Fix license (#8)
zR +1: osanseviero Verified signature- Fix license (cd899b1654a0d48cdc0361ab63a0e08689ee745d) Co-authored-by: Omar Sanseviero <osanseviero@users.noreply.huggingface.co>2844d07 -
for release
zRd055a49
-
arxiv public
zRa4101b3
-
update GPU memory to 24GB
zR4bbfb1d -
update_diffusers
zR0d6a353 -
Add usual tag for the task (#1)
zR +1: osanseviero Verified signature- Add usual tag for the task (82a6e38aeb9484dcb94ff379eda3de73beedbcb5) Co-authored-by: Omar Sanseviero <osanseviero@users.noreply.huggingface.co>95830a4 -
Update scheduler/scheduler_config.json (#2)
zR +1: YiYiXu Verified signature- Update scheduler/scheduler_config.json (00da947d39db0c720e501e2660442a4eb15ec60b) Co-authored-by: YiYi Xu <YiYiXu@users.noreply.huggingface.co>bf24ed3
-
GPU memory update
zR84c5dee -
1
zR7cdc618 -
update
zR15b8c5a -
init
Ubuntu57c28fa -
init
Ubuntu91597cf -
initial commit
zRVerified signatureb93f992