Skip to content

Adding WAN Lightning support #669

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
quic-amitraj merged 7 commits into from
Dec 20, 2025
Merged

Adding WAN Lightning support #669

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
5f0f28d
Updated WAN and added pytest
tv-karthikeya Dec 16, 2025
a0ee4f8
Refactor Wan pipeline, transformer
tv-karthikeya Dec 16, 2025
7a7605a
Refactor wan
tv-karthikeya Dec 18, 2025
9e43a58
Updating Jenkins file
tv-karthikeya Dec 19, 2025
501a052
Refactored to address review comments
tv-karthikeya Dec 19, 2025
54e9b2c
Merge branch 'main' into updated_wan_dit
tv-karthikeya Dec 19, 2025
42d0962
Increasing Diffusion test timeout
tv-karthikeya Dec 19, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions QEfficient/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@
)
from QEfficient.compile.compile_helper import compile
from QEfficient.diffusers.pipelines.flux.pipeline_flux import QEffFluxPipeline
from QEfficient.diffusers.pipelines.wan.pipeline_wan import QEffWanPipeline
from QEfficient.exporter.export_hf_to_cloud_ai_100 import qualcomm_efficient_converter
from QEfficient.generation.text_generation_inference import cloud_ai_100_exec_kv
from QEfficient.peft import QEffAutoPeftModelForCausalLM
Expand All @@ -55,6 +56,7 @@
"QEFFAutoModelForSpeechSeq2Seq",
"QEFFCommonLoader",
"QEffFluxPipeline",
"QEffWanPipeline",
]


Expand Down
2 changes: 1 addition & 1 deletion QEfficient/diffusers/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,7 @@ pip install dist/qefficient-0.0.1.dev0-py3-none-any.whl

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

---

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

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

## 📞 Support
Expand Down
Loading
Loading