Cseti's picture
Create README.md
8f1c4a1 verified
|
raw
history blame
1.2 kB
metadata
base_model:
  - THUDM/CogVideoX-5b
tags:
  - LoRA

You can use the finetuned model for inference with the following code: import torch from diffusers import CogVideoXPipeline from diffusers import export_to_video

import torch
from diffusers import CogVideoXPipeline
from diffusers import export_to_video

pipe = CogVideoXPipeline.from_pretrained(
    "THUDM/CogVideoX-5b", torch_dtype=torch.bfloat16
).to("cuda")

pipe.load_lora_weights("my-awesome-name/my-awesome-lora", adapter_name=["cogvideox-lora"])
pipe.set_adapters(["cogvideox-lora"], [1.0])

video = pipe("<my-awesome-prompt>").frames[0]
export_to_video(video, "output.mp4", fps=8)

This CogVideoX LoRA is created as part of a fan project for research purposes only and is not intended for commercial use. It is based on Wallace and Gromit film clips, which are protected by copyright. Users utilize the model at their own risk. Users are obligated to comply with copyright laws and applicable regulations. The Product has been developed for non-commercial purposes, and it is not my intention to infringe on any copyright. I assume no responsibility for any damages or legal consequences arising from the use of the Product.