mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-10-16 21:01:14 +08:00
[build]: flatten ktransformers package shim (#1955)
This commit is contained in:
@@ -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 @@ from pathlib import Path
|
||||
|
||||
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"]
|
||||
+3
-2
@@ -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"]
|
||||
|
||||
Reference in New Issue
Block a user