Instructions to use zai-org/CogVideoX-5b with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use zai-org/CogVideoX-5b 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-5b", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Inference
- Notebooks
- Google Colab
- Kaggle
Update transformers version
#1
by lysandre HF Staff - opened
README.md
CHANGED
|
@@ -207,7 +207,7 @@ optimizations and conversions to get a better experience.**
|
|
| 207 |
|
| 208 |
```shell
|
| 209 |
# diffusers>=0.30.1
|
| 210 |
-
# transformers>=
|
| 211 |
# accelerate>=0.33.0 (suggest install from source)
|
| 212 |
# imageio-ffmpeg>=0.5.1
|
| 213 |
pip install --upgrade transformers accelerate diffusers imageio-ffmpeg
|
|
|
|
| 207 |
|
| 208 |
```shell
|
| 209 |
# diffusers>=0.30.1
|
| 210 |
+
# transformers>=4.44.2
|
| 211 |
# accelerate>=0.33.0 (suggest install from source)
|
| 212 |
# imageio-ffmpeg>=0.5.1
|
| 213 |
pip install --upgrade transformers accelerate diffusers imageio-ffmpeg
|