Skip to content

Adding Vae Decoder in Wan#688

Merged
vbaddi merged 10 commits intoquic:mainfrom
mohiso22:diffuser
Jan 9, 2026
Merged

Adding Vae Decoder in Wan#688
vbaddi merged 10 commits intoquic:mainfrom
mohiso22:diffuser

Conversation

@mohiso22
Copy link
Copy Markdown
Contributor

No description provided.

@mohiso22 mohiso22 marked this pull request as ready for review December 23, 2025 07:04
Comment thread examples/diffusers/wan/wan_config.json Outdated
Comment thread QEfficient/diffusers/models/autoencoders/autoencoder_kl_wan.py Outdated
Comment thread QEfficient/diffusers/models/autoencoders/autoencoder_kl_wan.py
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_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/wan/pipeline_wan.py Outdated
Comment thread QEfficient/diffusers/pipelines/pipeline_module.py Outdated
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py Outdated
Comment thread QEfficient/diffusers/models/pytorch_transforms.py Outdated
@quic-amitraj
Copy link
Copy Markdown
Contributor

Refer flux pipeline for the changes and removed not needed class.

Comment thread QEfficient/diffusers/pipelines/pipeline_module.py Outdated
Comment thread QEfficient/diffusers/pipelines/wan/pipeline_wan.py Outdated
@quic-amitraj
Copy link
Copy Markdown
Contributor

To handle two different types of get_onnx_params for image and video pipelines, let's have two different methods like this:
def get_onnx_image_params(self)

def get_onnx_video_params(self)

Later inside init of pipeline map with the desired method with get_onnx_params like this-
self.vae_decode.get_onnx_params= self.vae_decode.get_onnx_image_params

@quic-amitraj
Copy link
Copy Markdown
Contributor

To handle two different types of get_onnx_params for image and video pipelines, let's have two different methods like this: def get_onnx_image_params(self)

def get_onnx_video_params(self)

Later inside init of pipeline map with the desired method with get_onnx_params like this- self.vae_decode.get_onnx_params= self.vae_decode.get_onnx_image_params

update flux init as well to reflect this changes.

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/pipeline_module.py Outdated
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

Mohit Soni and others added 8 commits January 7, 2026 11:34
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
@vbaddi vbaddi merged commit c57392d into quic:main Jan 9, 2026
4 checks passed
tv-karthikeya added a commit to tv-karthikeya/efficient-transformers that referenced this pull request Jan 9, 2026
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Co-authored-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Co-authored-by: vtirumal <vtirumal@qti.qualcomm.com>
tchawada pushed a commit to tchawada/QEff_tanisha that referenced this pull request Feb 4, 2026
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Co-authored-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Co-authored-by: vtirumal <vtirumal@qti.qualcomm.com>
tchawada pushed a commit to tchawada/QEff_tanisha that referenced this pull request Feb 4, 2026
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Co-authored-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Co-authored-by: vtirumal <vtirumal@qti.qualcomm.com>
tchawada pushed a commit to tchawada/QEff_tanisha that referenced this pull request Feb 4, 2026
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Co-authored-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Co-authored-by: vtirumal <vtirumal@qti.qualcomm.com>
qcdipankar pushed a commit to qcdipankar/efficient-transformers that referenced this pull request Feb 8, 2026
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Co-authored-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Co-authored-by: vtirumal <vtirumal@qti.qualcomm.com>
tchawada pushed a commit to tchawada/QEff_tanisha that referenced this pull request Feb 16, 2026
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Co-authored-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Co-authored-by: vtirumal <vtirumal@qti.qualcomm.com>
smedhe pushed a commit to smedhe/QEff_Sharvari that referenced this pull request Mar 8, 2026
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Co-authored-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Co-authored-by: vtirumal <vtirumal@qti.qualcomm.com>
Signed-off-by: Sharvari Medhe <smedhe@qti.qualcomm.com>
smedhe pushed a commit to smedhe/QEff_Sharvari that referenced this pull request Mar 8, 2026
Signed-off-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Signed-off-by: vtirumal <vtirumal@qti.qualcomm.com>
Co-authored-by: Mohit Soni <mohisoni@qti.qualcomm.com>
Co-authored-by: vtirumal <vtirumal@qti.qualcomm.com>
Signed-off-by: Sharvari Medhe <smedhe@qti.qualcomm.com>

Signed-off-by: Sharvari Medhe <smedhe@qti.qualcomm.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants