Skip to content
Open
Show file tree
Hide file tree
Changes from 8 commits
Commits
Show all changes
21 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
5227937
Merge branch 'main' into teacache-flux
LawJarp-A Nov 17, 2025
296aa8d
Merge branch 'main' into teacache-flux
LawJarp-A Nov 26, 2025
44663de
update it to make it model agnostic
LawJarp-A Nov 26, 2025
4d34020
add TeaCache hook tests and ensure cache integration passes style and…
LawJarp-A Nov 26, 2025
9dab52f
Add multi-model TeaCache support for Mochi, Lumina2, and CogVideoX wi…
LawJarp-A Dec 2, 2025
4cb71d6
simplify model extract; use logger
LawJarp-A Dec 8, 2025
ad90044
Merge branch 'main' into teacache-flux
LawJarp-A Dec 8, 2025
f2793fd
Merge branch 'main' into teacache-flux
LawJarp-A Dec 9, 2025
f0abb3c
fix(teacache): fix return_dict handling, CogVideoX fallback bug, add …
LawJarp-A Dec 10, 2025
4a6afef
Fix TeaCache state management and add num_inference_steps param
LawJarp-A Dec 10, 2025
8646908
fixed counter manage, cogvoideox missing norm proj added
LawJarp-A Dec 10, 2025
c62ddc2
Merge branch 'main' into teacache-flux
LawJarp-A Dec 11, 2025
c2c4c76
Merge branch 'huggingface:main' into teacache-flux
LawJarp-A Dec 16, 2025
38effa1
Refactor TeaCache hook into adapters
LawJarp-A Dec 16, 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
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
Loading