TensorRT-LLMs/requirements.txt
石晓伟 59f41c067d
Update TensorRT-LLM (#708)
* Update TensorRT-LLM

* update

* Bump version to 0.7.0
2023-12-20 16:38:28 +08:00

18 lines
201 B
Plaintext

accelerate==0.20.3
build
colored
cuda-python==12.2.0
diffusers==0.15.0
lark
mpi4py
numpy
onnx>=1.12.0
polygraphy
sentencepiece>=0.1.99
tensorrt>=8.6.0
torch
transformers==4.33.1
wheel
optimum
evaluate