Skip to content

Commit d2d6d83

Browse files
committed
Update the readme
Signed-off-by: Jingyu Xin <jingyux@nvidia.com>
1 parent 5ba76ac commit d2d6d83

1 file changed

Lines changed: 2 additions & 10 deletions

File tree

examples/diffusers/README.md

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -305,9 +305,6 @@ config = {
305305
"sparse_cfg": {
306306
"calibration": {
307307
"target_sparse_ratio": {"prefill": 0.5},
308-
"threshold_trials": [1e-6, 5e-6, 1e-5, 5e-5, 1e-4, 5e-4, 1e-3, 5e-3,
309-
1e-2, 2e-2, 5e-2, 1e-1, 2e-1, 3e-1, 5e-1, 7e-1,
310-
8e-1, 9e-1, 9.9e-1],
311308
},
312309
"*.attn1": {
313310
"method": "triton_skip_softmax",
@@ -339,15 +336,10 @@ output = pipeline(prompt="a dog on the beach", ...)
339336
The 14B model automatically sparsifies both `transformer` and `transformer_2`.
340337

341338
```bash
342-
# 5B model — calibrate + generate (4 prompts from OpenVid-1M, 151 frames, 40 steps)
343-
python sparsity/wan22_skip_softmax.py \
344-
--model-path Wan-AI/Wan2.2-TI2V-5B-Diffusers \
345-
--calibrate --target-sparsity 0.5 --calib-size 4 \
346-
--prompt "A sunset over mountains" --output out.mp4
347339

348-
# 14B model (both transformers sparsified)
340+
# 5B/14B model
349341
python sparsity/wan22_skip_softmax.py \
350-
--model-path Wan-AI/Wan2.2-T2V-A14B-Diffusers \
342+
--model-path Wan-AI/Wan2.2-T2V-A14B-Diffusers|Wan-AI/Wan2.2-TI2V-5B-Diffusers \
351343
--calibrate --target-sparsity 0.5 --calib-size 4 \
352344
--prompt "A sunset over mountains" --output out.mp4
353345
```

0 commit comments

Comments
 (0)