Views
No views yet
Packed ternary version available. Thepacked/folder now contains the real 2-bit packed transformer (1.78x smaller: 5.5 GB vs 9.8 GB FP16) that runs on theternary-quantpacked runtime. Load it withpacked/load_packed_dit.py(pip install ternary-quant). The root files remain FP16 so stockdiffusersstill works. Quality note: the packed transformer was quantized without activation calibration (memory-minimal streaming), so treat it as experimental.
| Property | Value |
|---|---|
| Base Model | Wan-AI/Wan2.2-TI2V-5B-Diffusers |
| Architecture | WanTransformer3DModel (DiT) |
| Transformer Params | 5.00B |
| Quantization | tritplane3 (306 linear layers) |
| Text Encoder (UMT5-XXL) | FP16 (preserved) |
| VAE (WanVAE) | FP16 (preserved) |
| License | Apache 2.0 |
| Method | Transformer Size |
|---|---|
| FP16 (original) | 10.02 GB |
| Ternary tritplane3 (theoretical packed) | ~5.0 GB |
| In this repo (dequantized FP16) | 9.4 GB |
1import torch
2from diffusers import WanPipeline
3from diffusers.utils import export_to_video
4
5pipe = WanPipeline.from_pretrained(
6 "AsadIsmail/Wan2.2-TI2V-5B-ternary",
7 torch_dtype=torch.bfloat16,
8)
9pipe.to("mps") # or "cuda"
10
11output = pipe(
12 prompt="a cat walking on green grass",
13 num_frames=81,
14 num_inference_steps=30,
15).frames[0]
16export_to_video(output, "output.mp4", fps=16)