[Perf][DreamZero]dreamzero supports CUDA graph, torch.compile, and DiT Caching. (#4213)
Signed-off-by: amy-why-3459 <wuhaiyan17@huawei.com>
This commit is contained in:
1 parent
63ba11b8b1
commit
f05e03aea8
22 files changed
+1204
-177
No files matched your search
Vendored
+2
@@ -13,6 +13,8 @@ Main cache backend implementations:
|
||||
the cache-dit library. Inherits from CacheBackend. Used via cache_backend="cache_dit".
|
||||
2. TeaCacheBackend: Hook-based backend for TeaCache acceleration. Inherits from
|
||||
CacheBackend. Used via cache_backend="tea_cache".
|
||||
3. StepCacheBackend: Velocity cosine step-skipping for DreamZero. Inherits from
|
||||
CacheBackend. Used via cache_backend="step_cache".
|
||||
|
||||
All backends implement the same interface:
|
||||
- enable(pipeline): Enable cache on the pipeline
|
||||
|
||||
Reference in new issue
Block a user