Nick Cao e6466cf0e6 [Refactor] Replace and ban a few torch.cuda functions in favor of torch.accelerator replacements. (#3365) ...

Signed-off-by: Nick Cao <ncao@redhat.com>

2026-05-07 23:23:59 +08:00

..

__init__.py

[Diffusion] Add cache-dit and unify diffusion cache backend interface (#250)

2025-12-15 15:10:45 +08:00

backend.py

[Misc] Clean up method name in BAGEL. (#2501)

2026-04-09 14:55:15 +08:00

coefficient_estimator.py

[Refactor] Replace and ban a few torch.cuda functions in favor of torch.accelerator replacements. (#3365)

2026-05-07 23:23:59 +08:00

config.py

Add Teacache Support for LongCat Image (#1487)

2026-04-18 07:52:35 +08:00

extractors.py

[Feature]: support Flux.1-dev tea_cache (#2774)

2026-04-26 23:55:16 +08:00

hook.py

[Feat] support TeaCache for Flux2 klein (#1234)

2026-03-23 10:19:25 +08:00

state.py

[Chore] Bump Python version and apply formatting changes for v0.12 release (#126)

2025-12-16 15:19:40 +08:00