import torch
from diffusers import DiffusionPipeline
from diffusers.utils import export_to_video
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("THUDM/CogVideoX-2b", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("BinaryLight1011/Cinemotion")
prompt = "A man with short gray hair plays a red electric guitar."
output = pipe(prompt=prompt).frames[0]
export_to_video(output, "output.mp4")- Downloads last month
- 15
Model tree for BinaryLight1011/Cinemotion
Base model
zai-org/CogVideoX-2b