diff --git a/ktransformers/__init__.py b/ktransformers.py
similarity index 68%
rename from ktransformers/__init__.py
rename to ktransformers.py
index df81f12a..a8ecd92a 100644
--- a/ktransformers/__init__.py
+++ b/ktransformers.py
@@ -1,4 +1,4 @@
-"""Top-level Python package for KTransformers.
+"""Top-level Python module for KTransformers.
 
 The runtime kernels live in kt-kernel. Optional SFT support is activated
 via pip install "ktransformers[sft]" which adds transformers-kt and
@@ -13,12 +13,13 @@
 
 def _read_repo_version() -> str:
     ns: dict[str, str] = {}
-    exec((Path(__file__).resolve().parents[1] / 'version.py').read_text(), ns)
-    return ns['__version__']
+    version_file = Path(__file__).resolve().with_name("version.py")
+    exec(version_file.read_text(), ns)
+    return ns["__version__"]
 
 
 try:
-    __version__ = version('ktransformers')
+    __version__ = version("ktransformers")
 except PackageNotFoundError:
     __version__ = _read_repo_version()
 
@@ -31,4 +32,4 @@ def has_sft_support() -> bool:
     return True
 
 
-__all__ = ['__version__', 'has_sft_support']
+__all__ = ["__version__", "has_sft_support"]
diff --git a/pyproject.toml b/pyproject.toml
index 9ff862bb..d48b6b5e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,5 +19,6 @@ classifiers = [
 Homepage = "https://github.com/kvcache-ai/ktransformers"
 
 [tool.setuptools]
-# Ship a minimal top-level Python package so the distribution is importable.
-packages = ["ktransformers"]
+# Ship a minimal top-level Python module so the distribution is importable
+# without keeping a parallel source package directory at the repository root.
+py-modules = ["ktransformers"]