Skip to content

Adding WAN Lightning support#669

Merged
quic-amitraj merged 7 commits intoquic:mainfrom
quic-amitraj:updated_wan_dit
Dec 20, 2025
Merged

Adding WAN Lightning support#669
quic-amitraj merged 7 commits intoquic:mainfrom
quic-amitraj:updated_wan_dit

Conversation

@tv-karthikeya
Copy link
Copy Markdown
Contributor

@tv-karthikeya tv-karthikeya commented Dec 16, 2025

Model: lightx2v/Wan2.2-Lightning

Support Wan Unified Transformer on QAIC

Comment thread pyproject.toml Outdated
Comment thread QEfficient/diffusers/models/transformers/transformer_wan.py Outdated
Comment thread QEfficient/diffusers/pipelines/pipeline_utils.py Outdated
Comment thread QEfficient/diffusers/pipelines/configs/wan_config.json Outdated
Comment thread QEfficient/diffusers/pipelines/pipeline_module.py Outdated
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py Outdated
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py Outdated
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py Outdated
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py Outdated
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py Outdated
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py Outdated
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py
@quic-amitraj
Copy link
Copy Markdown
Contributor

Overall suggestion is to refer the flux pipeline and make changes based on that.

Comment thread examples/diffusers/wan/wan_config.json Outdated
Comment thread tests/diffusers/wan_test_config.json Outdated
Comment thread QEfficient/utils/constants.py
Comment thread QEfficient/diffusers/pipelines/pipeline_utils.py Outdated
Comment thread QEfficient/diffusers/pipelines/configs/wan_config.json Outdated
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
  -  Moved blocking to utils from transformer to make it generic
  -  Removed usage of register_modules, components and refactored pipeline, test
  -  Adding Readme, few more constants

Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
  -  Addressed spec file issue
  -  Adding compile only flag by default
  -  Refactor of common utils

Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Comment thread pyproject.toml
Comment thread tests/diffusers/wan_test_config.json Outdated
Comment thread tests/diffusers/test_wan.py Outdated
Comment thread QEfficient/diffusers/pipelines/pipeline_module.py Outdated
Comment thread QEfficient/diffusers/pipelines/pipeline_module.py Outdated
Comment thread QEfficient/diffusers/pipelines/pipeline_module.py Outdated
Comment thread QEfficient/diffusers/pipelines/pipeline_utils.py Outdated
Comment thread QEfficient/diffusers/models/modeling_utils.py
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py Outdated
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py Outdated
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py Outdated
Comment thread QEfficient/diffusers/pipelines/configs/wan_config.json Outdated
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Signed-off-by: Karthikeya <vtirumal@qti.qualcomm.com>
Copy link
Copy Markdown
Contributor

@quic-amitraj quic-amitraj left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM now

Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
@quic-amitraj quic-amitraj merged commit 8721324 into quic:main Dec 20, 2025
4 checks passed
quic-amitraj pushed a commit that referenced this pull request Dec 22, 2025
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>
abhishek-singh591 pushed a commit to abhishek-singh591/quic_abhishek that referenced this pull request Jan 2, 2026
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>
Signed-off-by: abhishek-singh591 <sabhis@qti.qualcomm.com>
quic-dhirajku pushed a commit to quic-dhirajku/efficient-transformers that referenced this pull request Jan 2, 2026
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>
Signed-off-by: Dhiraj Kumar Sah <dhirajku@qti.qualcomm.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants