Instructions to use zai-org/CogVideoX-5b-I2V with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use zai-org/CogVideoX-5b-I2V with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline from diffusers.utils import load_image, export_to_video # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("zai-org/CogVideoX-5b-I2V", dtype=torch.bfloat16, device_map="cuda") pipe.to("cuda") prompt = "A man with short gray hair plays a red electric guitar." image = load_image( "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png" ) output = pipe(image=image, prompt=prompt).frames[0] export_to_video(output, "output.mp4") - Notebooks
- Google Colab
- Kaggle
Improve model card with pipeline tag and library name
#17
by nielsr HF Staff - opened
This PR adds the pipeline_tag and library_name to the model card metadata. The pipeline_tag is set to text-to-video to accurately reflect the model's capabilities. The library_name is set to diffusers based on the provided usage examples. This improves the discoverability and usability of the model. The license has been updated to reflect the information from the GitHub README.