From 605d1de013ab51ad4f12d2d6ab18c55c76585cea Mon Sep 17 00:00:00 2001 From: tzhouam Date: Fri, 24 Oct 2025 17:54:19 +0800 Subject: [PATCH] Update entry points in vllm_omni module - Reintroduced AsyncOmniLLM to the main entry points. - Added import for patch module. - Cleaned up comments regarding AsyncOmniLLM support. --- vllm_omni/__init__.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/vllm_omni/__init__.py b/vllm_omni/__init__.py index 4281a4e55..a8bb692c5 100644 --- a/vllm_omni/__init__.py +++ b/vllm_omni/__init__.py @@ -15,9 +15,7 @@ Architecture: __version__ = "0.1.0" __author__ = "vLLM-omni Team" __email__ = "hsliuustc@gmail.com" - -# Main entry points -from .entrypoints.omni_llm import OmniLLM +from . import patch from .config import ( DiTCacheConfig, DiTConfig, @@ -27,7 +25,7 @@ from .config import ( ) # Main entry points -from .entrypoints.omni_llm import OmniLLM #AsyncOmniLLM, +from .entrypoints.omni_llm import AsyncOmniLLM, OmniLLM __all__ = [ # Version info @@ -36,8 +34,7 @@ __all__ = [ "__email__", # Main components "OmniLLM", - #AsyncOmniLLM is not supported yet - + "AsyncOmniLLM", # Configuration "OmniStageConfig", "DiTConfig",