import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Wan-AI/Wan2.2-TI2V-5B", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("ostris/wan22_5b_i2v_crush_it_lora")
prompt = "crush it"
image = pipe(prompt).images[0]Wan 2.2 5B I2V Crush It LoRA
- Prompt
- crush it
Model description
This is an Image to Video LoRA trained while filming a tutorial with AI Toolkit. That tutorial video can be found here: How to Train a Wan 2.2 5B Image to Video LoRA With AI Toolkit
Trigger words
You should use crush it to trigger the image generation.
Download model
Download them in the Files & versions tab.
- Downloads last month
- 61