How to use Wan-AI/Wan2.1-T2V-1.3B with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("Wan-AI/Wan2.1-T2V-1.3B", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0]
Can I use this model in CPU-Only mode?
probably resource intensive, cpu only mya be a bottleneck, unless a dedicated cpu like a mac mini or sth. My 2 cents
· Sign up or log in to comment
