thu-ml / thu-ml/TurboDiffusion

Can TurboDiffusion be applied to Autoregressive generation in a training-free manner?

Open
#81 1 comment 0 reactions 0 assignees View on GitHub

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:

  1. 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)?

  2. 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

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.