Skip to content

Commit 80b26eb

Browse files
tv-karthikeyaquic-amitraj
authored andcommitted
Adding WAN Lightning support (#669)
Model: lightx2v/Wan2.2-Lightning Support Wan Unified Transformer on QAIC --------- Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com> Signed-off-by: Karthikeya <vtirumal@qti.qualcomm.com>
1 parent 0c38085 commit 80b26eb

22 files changed

+3047
-51
lines changed

QEfficient/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
)
3030
from QEfficient.compile.compile_helper import compile
3131
from QEfficient.diffusers.pipelines.flux.pipeline_flux import QEffFluxPipeline
32+
from QEfficient.diffusers.pipelines.wan.pipeline_wan import QEffWanPipeline
3233
from QEfficient.exporter.export_hf_to_cloud_ai_100 import qualcomm_efficient_converter
3334
from QEfficient.generation.text_generation_inference import cloud_ai_100_exec_kv
3435
from QEfficient.peft import QEffAutoPeftModelForCausalLM
@@ -55,6 +56,7 @@
5556
"QEFFAutoModelForSpeechSeq2Seq",
5657
"QEFFCommonLoader",
5758
"QEffFluxPipeline",
59+
"QEffWanPipeline",
5860
]
5961

6062

QEfficient/diffusers/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ pip install dist/qefficient-0.0.1.dev0-py3-none-any.whl
6262

6363
## 🎯 Supported Models
6464
-[`black-forest-labs/FLUX.1-schnell`](https://huggingface.co/black-forest-labs/FLUX.1-schnell)
65+
-[`lightx2v/Wan2.2-Lightning`](https://huggingface.co/lightx2v/Wan2.2-Lightning)
6566

6667
---
6768

@@ -83,7 +84,6 @@ We welcome contributions! Please see our [Contributing Guide](../../CONTRIBUTING
8384
## 🙏 Acknowledgments
8485

8586
- **HuggingFace Diffusers**: For the excellent foundation library
86-
- **Stability AI**: For the amazing Stable Diffusion models
8787
---
8888

8989
## 📞 Support

0 commit comments

Comments
 (0)