Skip to content

vllm_omni.diffusion.cache.selector

get_cache_backend

get_cache_backend(
    cache_backend: str | None, cache_config: Any
) -> CacheBackend | None

Get cache backend instance based on cache_backend string.

This is a selector function that routes to the appropriate backend implementation. - cache_dit: Uses CacheDiTBackend with enable()/refresh() interface - tea_cache: Uses TeaCacheBackend with enable()/refresh() interface - mag_cache: Uses MagCacheBackend with enable()/refresh() interface

Parameters:

Name Type Description Default
cache_backend str | None

Cache backend name ("cache_dit", "tea_cache", "mag_cache", or None).

required
cache_config Any

Cache configuration (dict or DiffusionCacheConfig instance).

required

Returns:

Type Description
CacheBackend | None

Cache backend instance (CacheDiTBackend, TeaCacheBackend, or MagCacheBackend)

CacheBackend | None

if cache_backend is set, None otherwise.

Raises:

Type Description
ValueError

If cache_backend is unsupported.