mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-04-28 03:39:48 +00:00
29 lines
671 B
Python
29 lines
671 B
Python
"""Lightweight top-level package: pip install ktransformers -> installs kt-kernel.
|
|
|
|
Extras:
|
|
- ktransformers[sft] installs transformers-kt + accelerate-kt
|
|
- ktransformers[sglang] installs sglang-kt
|
|
"""
|
|
from pathlib import Path
|
|
from setuptools import setup
|
|
|
|
_version_file = Path(__file__).resolve().parent / "version.py"
|
|
_ns = {}
|
|
exec(_version_file.read_text(), _ns)
|
|
_v = _ns["__version__"]
|
|
|
|
setup(
|
|
version=_v,
|
|
install_requires=[
|
|
f"kt-kernel=={_v}",
|
|
],
|
|
extras_require={
|
|
"sft": [
|
|
"transformers-kt==5.6.0",
|
|
"accelerate-kt==1.14.0",
|
|
],
|
|
"sglang": [
|
|
"sglang-kt>=0.5.3",
|
|
],
|
|
},
|
|
)
|