[refactor] teacache refactor (#395)

Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
This commit is contained in:
Jiangyun Zhu authored and GitHub committed 2025-12-22 06:49:04 +08:00
1 parent 01f28217b1
commit a7fe4fba4e
9 files changed
+94 -107

No files matched your search

-2
View File
@@ -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")