Diffusers How to use Luo-Yihong/TDM_CogVideoX-2B_LoRA with Diffusers:
pip install -U diffusers transformers accelerate
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("Luo-Yihong/TDM_CogVideoX-2B_LoRA")
prompt = "A man with short gray hair plays a red electric guitar."
output = pipe(prompt=prompt).frames[0]
export_to_video(output, "output.mp4") 
