[refactor] teacache refactor (#395)
Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
This commit is contained in:
1 parent
01f28217b1
commit
a7fe4fba4e
9 files changed
+94
-107
No files matched your search
@@ -146,8 +146,6 @@ class TestTeaCacheBackend:
|
||||
|
||||
# Verify hook was applied
|
||||
assert backend.enabled is True
|
||||
assert hasattr(mock_pipeline, "_cache_backend")
|
||||
assert mock_pipeline._cache_backend is backend
|
||||
mock_apply_hook.assert_called_once()
|
||||
|
||||
@patch("vllm_omni.diffusion.cache.teacache.backend.apply_teacache_hook")
|
||||
|
||||
Reference in new issue
Block a user