b690a10e3a
Signed-off-by: Lancer <maruixiang6688@gmail.com> Co-authored-by: 汪志鹏 <wangzhipeng628@gmail.com> Co-authored-by: Gao Han <hgaoaf@connect.ust.hk>
33 lines
822 B
Python
33 lines
822 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from vllm_omni.diffusion.cache.magcache.config import MagCacheConfig
|
|
from vllm_omni.diffusion.cache.magcache.hook import (
|
|
MagCacheBlockHook,
|
|
MagCacheHeadHook,
|
|
MagCacheState,
|
|
apply_mag_cache_hook,
|
|
)
|
|
from vllm_omni.diffusion.cache.magcache.strategy import (
|
|
Flux2MagCacheStrategy,
|
|
FluxMagCacheStrategy,
|
|
MagCacheStrategy,
|
|
MagCacheStrategyRegistry,
|
|
get_strategy,
|
|
register_strategy,
|
|
)
|
|
|
|
__all__ = [
|
|
"Flux2MagCacheStrategy",
|
|
"FluxMagCacheStrategy",
|
|
"MagCacheBlockHook",
|
|
"MagCacheConfig",
|
|
"MagCacheHeadHook",
|
|
"MagCacheState",
|
|
"MagCacheStrategy",
|
|
"MagCacheStrategyRegistry",
|
|
"apply_mag_cache_hook",
|
|
"get_strategy",
|
|
"register_strategy",
|
|
]
|