o ۷i@s^ddlmZddlmZddlmZddlmZddlm Z de dBded edBfd d Z dS) )Any) CacheBackend)CacheDiTBackend)TeaCacheBackend)DiffusionCacheConfig cache_backendN cache_configreturncCsX|dus|dkr dSt|trt|}|dkrt|S|dkr$t|Std|d)aGet 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 Args: cache_backend: Cache backend name ("cache_dit", "tea_cache", or None). cache_config: Cache configuration (dict or DiffusionCacheConfig instance). Returns: Cache backend instance (CacheDiTBackend or TeaCacheBackend) if cache_backend is set, None otherwise. Raises: ValueError: If cache_backend is unsupported. Nnone cache_dit tea_cachezUnsupported cache backend: z%. Supported: 'cache_dit', 'tea_cache') isinstancedictr from_dictrr ValueError)rrrX/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm_omni/diffusion/cache/selector.pyget_cache_backend s  r) typingrvllm_omni.diffusion.cache.baser+vllm_omni.diffusion.cache.cache_dit_backendr*vllm_omni.diffusion.cache.teacache.backendrvllm_omni.diffusion.datarstrrrrrrs    "