Skip to content

feat: add generic DiT support to spectrum cache#1336

Open
rmatif wants to merge 1 commit intoleejet:masterfrom
rmatif:dit-spectrum
Open

feat: add generic DiT support to spectrum cache#1336
rmatif wants to merge 1 commit intoleejet:masterfrom
rmatif:dit-spectrum

Conversation

@rmatif
Copy link
Contributor

@rmatif rmatif commented Mar 12, 2026

Implementing the paper literally is heavier because it depends on architecture-specific internal feature hooks and final-block forecasting inside each DiT family, which would require invasive model-side changes and separate validation per model. This patch keeps Spectrum generic at the sampler level

Comparaison with easycache with 20 steps:

Baseline EasyCache Spectrum
Image Baseline EasyCache Spectrum
Speedup x1.82 x1.82
Config 0.2, 0.15, 0.95 default

@JohnLoveJoy
Copy link

I'd say it's better than baseline. That ear looked like an artifact.

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