Skip to content

Implement TeaCache #12652

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 7 commits into
base: main
Choose a base branch
from
+660 −1
Open

Implement TeaCache #12652

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
781c7e3
base implement teacahce flux; follow original impl
LawJarp-A Nov 13, 2025
549bf97
update cache utils with teacache
LawJarp-A Nov 13, 2025
29d4ffc
change hook to inline transformer processing instead of calling origi…
LawJarp-A Nov 13, 2025
07c6718
add extensive docstring (auto gen); add repr
LawJarp-A Nov 14, 2025
a7598a1
add param validation and error messages
LawJarp-A Nov 14, 2025
d9648e5
add basic logging
LawJarp-A Nov 14, 2025
59cb890
add compatible test
LawJarp-A Nov 14, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions src/diffusers/hooks/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
from .context_parallel import apply_context_parallel
from .faster_cache import FasterCacheConfig, apply_faster_cache
from .first_block_cache import FirstBlockCacheConfig, apply_first_block_cache
from .flux_teacache import FluxTeaCacheConfig, apply_flux_teacache
from .group_offloading import apply_group_offloading
from .hooks import HookRegistry, ModelHook
from .layer_skip import LayerSkipConfig, apply_layer_skip
Expand Down
Loading