Skip to content

Added support of QEffDiffusionPipeline for Diffusers#977

Open
quic-amitraj wants to merge 1 commit into
quic:mainfrom
quic-amitraj:diffusers_infra_upgrade
Open

Added support of QEffDiffusionPipeline for Diffusers#977
quic-amitraj wants to merge 1 commit into
quic:mainfrom
quic-amitraj:diffusers_infra_upgrade

Conversation

@quic-amitraj
Copy link
Copy Markdown
Contributor

@quic-amitraj quic-amitraj commented May 11, 2026

Summary

Adds a generic QEffDiffusionPipeline entrypoint for diffusers that auto-dispatches to the correct concrete QEff pipeline, while moving shared infra to a common base.

What Changed

  • Added QEffDiffusionPipeline.from_pretrained(...) dispatch by HF pipeline class and return of concrete child objects:
    • QEffFluxPipeline
    • QEffWanPipeline
    • QEffWanImageToVideoPipeline
  • Centralized shared compile/export flow in generic base APIs.
  • Refactored child pipelines to inherit and reuse the shared infra.
  • Updated all diffusion example scripts (including custom + first-block-cache) to use generic factory creation.
  • Added consolidated unit tests in tests/unit_test/utils/test_diffusion_pipeline_infra.py.

@quic-amitraj quic-amitraj self-assigned this May 11, 2026
@quic-amitraj quic-amitraj added the Diffusers Use for PR related to diffusers in efficient-transformers. label May 11, 2026
@quic-amitraj quic-amitraj force-pushed the diffusers_infra_upgrade branch from a0c0be1 to 9b8e8b9 Compare May 11, 2026 20:50
Signed-off-by: Amit Raj <amitraj@qti.qualcomm.com>
@quic-amitraj quic-amitraj force-pushed the diffusers_infra_upgrade branch from 9b8e8b9 to c1097ba Compare May 18, 2026 10:23
@quic-amitraj quic-amitraj requested a review from vbaddi May 18, 2026 10:26
@quic-amitraj quic-amitraj force-pushed the diffusers_infra_upgrade branch from c1097ba to 8822fc0 Compare May 18, 2026 10:32
@quic-amitraj quic-amitraj marked this pull request as ready for review May 18, 2026 10:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Diffusers Use for PR related to diffusers in efficient-transformers.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant