From f6158e7576ab1d71f469484905a5caa6d22c19db Mon Sep 17 00:00:00 2001
From: tzhouam <tzhouam@connect.ust.hk>
Date: Fri, 24 Oct 2025 18:01:02 +0800
Subject: [PATCH] debug

---
 vllm_omni/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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",