diff --git a/vllm_omni/entrypoints/omni.py b/vllm_omni/entrypoints/omni.py
index 320a85805..02499525f 100644
--- a/vllm_omni/entrypoints/omni.py
+++ b/vllm_omni/entrypoints/omni.py
@@ -2,8 +2,8 @@
 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
 import os
 
-from vllm_omni.diffusion.omni_diffusion import OmniDiffusion
 from vllm_omni.diffusion.utils.hf_utils import is_diffusion_model
+from vllm_omni.entrypoints.omni_diffusion import OmniDiffusion
 from vllm_omni.entrypoints.omni_llm import OmniLLM
 
 
diff --git a/vllm_omni/diffusion/omni_diffusion.py b/vllm_omni/entrypoints/omni_diffusion.py
similarity index 100%
rename from vllm_omni/diffusion/omni_diffusion.py
rename to vllm_omni/entrypoints/omni_diffusion.py