Files
xc-llm-ascend/requirements.txt
2026-01-23 09:52:55 +08:00

38 lines
634 B
Plaintext

# Should be mirrored in pyporject.toml
cmake>=3.26
decorator
einops
numpy<2.0.0
packaging
pip
pybind11
pyyaml
scipy
pandas
setuptools>=64
setuptools-scm>=8
torch==2.9.0
torchvision
wheel
xgrammar>=0.1.30
pandas-stubs
opencv-python-headless<=4.11.0.86 # Required to avoid numpy version conflict with vllm
compressed_tensors>=0.11.0
# 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.9.0
arctic-inference==0.1.1
transformers>=4.57.3
fastapi<0.124.0
triton-ascend==3.2.0