Skip to content
Open
Show file tree
Hide file tree
Changes from all 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 @@ -25,4 +25,5 @@
from .layerwise_casting import apply_layerwise_casting, apply_layerwise_casting_hook
from .pyramid_attention_broadcast import PyramidAttentionBroadcastConfig, apply_pyramid_attention_broadcast
from .smoothed_energy_guidance_utils import SmoothedEnergyGuidanceConfig
from .teacache import TeaCacheConfig, apply_teacache
from .taylorseer_cache import TaylorSeerCacheConfig, apply_taylorseer_cache
Loading