35 lines
916 B
Python
35 lines
916 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from vllm_omni.diffusion.cache.magcache.backend import MagCacheBackend
|
|
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",
|
|
"MagCacheBackend",
|
|
"MagCacheConfig",
|
|
"MagCacheHeadHook",
|
|
"MagCacheState",
|
|
"MagCacheStrategy",
|
|
"MagCacheStrategyRegistry",
|
|
"apply_mag_cache_hook",
|
|
"get_strategy",
|
|
"register_strategy",
|
|
]
|