28 lines
410 B
Plaintext
28 lines
410 B
Plaintext
|
# Should be mirrored in pyporject.toml
|
||
|
|
cmake>=3.26
|
||
|
|
decorator
|
||
|
|
einops
|
||
|
|
numpy<2.0.0
|
||
|
|
packaging
|
||
|
|
pip
|
||
|
|
pybind11
|
||
|
|
pyyaml
|
||
|
|
scipy
|
||
|
|
setuptools>=64
|
||
|
|
setuptools-scm>=8
|
||
|
|
torch>=2.7.1
|
||
|
|
torchvision
|
||
|
|
wheel
|
||
|
|
|
||
|
|
# requirements for disaggregated prefill
|
||
|
|
msgpack
|
||
|
|
quart
|
||
|
|
|
||
|
|
# Required for N-gram speculative decoding
|
||
|
|
numba
|
||
|
|
|
||
|
|
# Install torch_npu
|
||
|
|
--pre
|
||
|
|
--extra-index-url https://mirrors.huaweicloud.com/ascend/repos/pypi
|
||
|
|
torch-npu==2.7.1.dev20250724
|