thu-ml / thu-ml/TurboDiffusion
Can TurboDiffusion be applied to Autoregressive generation in a training-free manner?
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 3.7k
- Forks
- 277
- Avg merge
- 2h 57m
- Merged PRs (30d)
- 2
Description
Great work on the project. I am interested in applying your acceleration techniques (SageSLA/SageAttention) to Autoregressive (AR) video generation models.
I have two specific questions regarding the adaptation:
-
Training-free vs. Fine-tuning Can the current SageSLA or SageAttention directly replace the attention mechanism in a pre-trained AR model without any training? Or is fine-tuning mandatory (e.g., to align the Linear Attention branch or adapt to the quantization)?
-
KV Cache Support For AR decoding, KV Cache is essential. Does your current INT8 implementation support INT8 KV Cache management to save memory during step-by-step generation?
I noticed in the roadmap/documentation that there are plans to extend TurboDiffusion to Autoregressive (AR) models. Could you share a rough estimate of when we might expect AR support to land?
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the project roadmap/documentation and the existing SageSLA/SageAttention INT8 implementation. Investigate whether the current attention replacements and KV-cache handling cover autoregressive decoding, then document the training requirements, INT8 KV-cache support, and expected scope or timing for AR support.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- ai, machine-learning, performance
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 20/100