[Chore]: Clean up pyproject.toml and cache-dit backend (#351)

Signed-off-by: Isotr0py <mozf@mail2.sysu.edu.cn>
This commit is contained in:
Isotr0py authored and GitHub committed 2025-12-18 04:03:53 +00:00
1 parent 1f4e1badc6
commit cf8d297af0
3 files changed
+2 -15

No files matched your search

+2 -12
View File
@@ -10,6 +10,8 @@ pipelines in vllm-omni, supporting both single and dual-transformer architecture
from collections.abc import Callable
from typing import Any, Optional
import cache_dit
from cache_dit import BlockAdapter, DBCacheConfig, ForwardPattern, ParamsModifier, TaylorSeerCalibratorConfig
from vllm.logger import init_logger
from vllm_omni.diffusion.cache.base import CacheBackend
@@ -17,15 +19,6 @@ from vllm_omni.diffusion.data import DiffusionCacheConfig, OmniDiffusionConfig
logger = init_logger(__name__)
try:
import cache_dit
from cache_dit import BlockAdapter, DBCacheConfig, ForwardPattern, ParamsModifier, TaylorSeerCalibratorConfig
CACHE_DIT_AVAILABLE = True
except ImportError:
CACHE_DIT_AVAILABLE = False
logger.warning("cache-dit is not installed. Cache-dit acceleration will not be available.")
# Registry of custom cache-dit enablers for specific models
# Maps pipeline names to their cache-dit enablement functions
@@ -325,9 +318,6 @@ class CacheDiTBackend(CacheBackend):
Args:
pipeline: The diffusion pipeline instance.
"""
if not CACHE_DIT_AVAILABLE:
logger.warning("cache-dit is not available, skipping cache-dit setup.")
return
# Extract pipeline name from pipeline
pipeline_name = pipeline.__class__.__name__