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. |