Origami WanLora

🚀 Origami Lora for WanVideo2.1

This is an Origami Lora for WanVideo2.1, which can generate videos based on specific prompts.

🚀 Quick Start

✨ Features

• Use the trigger word origami to generate videos.
• Compatible with the Diffusers library.
• Model weights are available in Safetensors format.

📦 Installation

pip install git+https://github.com/huggingface/diffusers.git

💻 Usage Examples

Basic Usage

import torch
from diffusers.utils import export_to_video
from diffusers import AutoencoderKLWan, WanPipeline
from diffusers.schedulers.scheduling_unipc_multistep import UniPCMultistepScheduler

# Available models: Wan-AI/Wan2.1-T2V-14B-Diffusers, Wan-AI/Wan2.1-T2V-1.3B-Diffusers
model_id = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
flow_shift = 5.0  # 5.0 for 720P, 3.0 for 480P
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=flow_shift)
pipe.to("cuda")

pipe.load_lora_weights("shauray/Origami_WanLora")

pipe.enable_model_cpu_offload() #for low-vram environments

prompt = "origami style bull charging towards a man"

output = pipe(
    prompt=prompt,
    height=480,
    width=720,
    num_frames=81,
    guidance_scale=5.0,
).frames[0]
export_to_video(output, "output.mp4", fps=16)

📚 Documentation

🔍 Trigger words

You should use origami to trigger the video generation.

📥 Download model

Weights for this model are available in Safetensors format. Download them in the Files & versions tab.

📄 License

This project is licensed under the Apache-2.0 license.

⚠️ Important Note

This Lora is not perfect. There is a little issue towards the bottom of every generation because the dataset had those problems (I messed up cleaning those).