Instructions to use Wan-AI/Wan2.1-T2V-14B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use Wan-AI/Wan2.1-T2V-14B 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-14B", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Inference
- Notebooks
- Google Colab
- Kaggle
Error while running the generate script, looks like some issue with model loading.. can someone please point to the right checkpoint ?
#28
by harshil12432 - opened
[2025-03-20 03:19:44,611] INFO: loading /dev/shm/model/models_t5_umt5-xxl-enc-bf16.pth
/.conda/envs/py39/lib/python3.9/site-packages/torch/serialization.py", line 1628, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, '='.