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")
zR +1: a-r-r-o-w Verified signature- Upload VAE in fp32 (567fb6a18b1ae8dbb9fcf29c3ea1f17de05bd9ed) - Update vae/config.json (f03dce9522087167fb0c9a431976e46871f2df80) Co-authored-by: Aryan V S <a-r-r-o-w@users.noreply.huggingface.co>a6f0f48