Add TurboT2AV SageSLA, FastNorm, and W8A8 acceleration submodule#132
Open
liuyuxiang1021 wants to merge 11 commits into
Open
Add TurboT2AV SageSLA, FastNorm, and W8A8 acceleration submodule#132liuyuxiang1021 wants to merge 11 commits into
liuyuxiang1021 wants to merge 11 commits into
Conversation
85baa11 to
19dff3a
Compare
19dff3a to
8972961
Compare
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
This PR adds TurboT2AV as an optional TurboDiffusion submodule for accelerated text-to-audio-video inference.
The submodule
mainbranch is based on the cleanorigin_inferencebranch and contains inference-only integration:Training scripts, training workflow changes, and trained SLA adapter checkpoints are not included.
Defaults
The original inference behavior remains the CLI default:
--attention_type default--fast_normdisabled--quant_lineardisabled--sla_topk 1.0when SLA/SageSLA is selected without an explicit ratioThe recommended accelerated command is opt-in and uses self-attention SageSLA with
topk=0.3, FastNorm, and TileLang W8A8 Linear.Cumulative H20 validation
The breakdown follows TurboDiffusion's cumulative order: optimize the teacher with W8A8/FastNorm, switch to the rCM-distilled student while retaining those optimizations, and finally add SageSLA. Generator-only measurements use one NVIDIA H20,
1024x1792, 121 frames, and--skip_decode.topk=0.3The long teacher rows use one measured generation; student rows report medians from repeated generations. The non-cumulative pure 4-step student takes 16.1096s/video, making the final path 2.75x faster than the pure student as well.
At this resolution the video latent contains 28,672 self-attention tokens. Component checks measured SageSLA self-attention at 4.82x over SDPA and the two dominant TileLang W8A8 FFN GEMM shapes at 1.56x and 1.61x over BF16.
topk=0.3is the documented high-resolution speed/quality tradeoff based on decoded prompt comparisons. SageSLA is approximate sparse attention, so lower ratios can change generated content more visibly.Submodule
liuyuxiang1021/turbo-t2avmainb57fb9bbbe92324b1aebc5b7e5b17ded3538aa8dorigin_inferenceatc99f03b7b615661f63513b3816ea6c62b754c5ceThe parent
.gitmodulesentry tracks the submodule'smainbranch, while the gitlink pins the exact commit above.Validation
git diff --check