Skip to content

[Feat]:Accelerate LightX2V using magi_compiler#1

Merged
jiahy0825 merged 4 commits into
mainfrom
dev/magi_compiler
Mar 25, 2026
Merged

[Feat]:Accelerate LightX2V using magi_compiler#1
jiahy0825 merged 4 commits into
mainfrom
dev/magi_compiler

Conversation

@wtr0504
Copy link
Copy Markdown
Collaborator

@wtr0504 wtr0504 commented Mar 25, 2026

  • Integrated MagiCompiler: Enabled inference acceleration on LightX2V for the Huanyuan1.5_480P_t2v_distill and Wan2.2 TI2V-5B models.
  • Refactored TransformerInfer: Replaced certain components with native PyTorch operators to improve compatibility with Inductor for better performance. Also adjusted tensor shape transformations to make them more recognizable by TorchDynamo.
  • Optimized ulysses_attn module: Modified parts of the implementation to eliminate graph breaks and improve compilation stability.
  • Updated README.md: Added installation instructions for MagiCompiler along with detailed usage guidelines.

chore

fix graph break in ulysses_attn

Update README and requirements

Update MagiCompiler Usage

chore

Update readme

Fix transformer infer torch defaults and shape handling; update MagiCompiler install docs

rm redundant code

chore
@wtr0504 wtr0504 force-pushed the dev/magi_compiler branch from 70596c5 to 7cd0c8e Compare March 25, 2026 03:06
Copy link
Copy Markdown
Collaborator

@jiahy0825 jiahy0825 left a comment

Choose a reason for hiding this comment

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

LGTM

@jiahy0825 jiahy0825 merged commit 3ed48d6 into main Mar 25, 2026
2 checks passed
@jiahy0825 jiahy0825 deleted the dev/magi_compiler branch March 25, 2026 03:53
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.

2 participants