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
diffusion_pytorch_model.bin missing?
#26
by Rafael1 - opened
I am trying to find the diffusion_pytorch_model.bin for a Confyui workflow but I cant find it. Any ideas?
Thank you!
I'm not quite clear on how ComfyUI reads the model(We did not Provide ComfyUI Method), but our model uses safetensor, and I'm not sure why it reads the .bin file, as there is an index file under the safetensor file.