Diffusers How to use BelGio13/cogvideoX-I2V-locobot 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("THUDM/CogVideoX-5b-I2V", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("BelGio13/cogvideoX-I2V-locobot")
prompt = "Realistic motion, smooth, complete, high resolution"
input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png")
output = pipe(image=input_image, prompt=prompt).frames[0]
export_to_video(output, "output.mp4") 
