diff --git a/vllm_omni/__init__.py b/vllm_omni/__init__.py index a8bb692c5..94f3cf9b8 100644 --- a/vllm_omni/__init__.py +++ b/vllm_omni/__init__.py @@ -25,7 +25,7 @@ from .config import ( ) # Main entry points -from .entrypoints.omni_llm import AsyncOmniLLM, OmniLLM +from .entrypoints.omni_llm import OmniLLM __all__ = [ # Version info @@ -34,7 +34,7 @@ __all__ = [ "__email__", # Main components "OmniLLM", - "AsyncOmniLLM", + # AsyncOmniLLM is not supported yet # Configuration "OmniStageConfig", "DiTConfig",