TensorRT-LLMs/examples/quantization/requirements.txt
石晓伟 8f91cff22e
TensorRT-LLM Release 0.15.0 (#2529)
Co-authored-by: Kaiyu Xie <26294424+kaiyux@users.noreply.github.com>
2024-12-04 13:44:56 +08:00

8 lines
152 B
Plaintext

tensorrt_llm==0.15.0
datasets>=2.14.4
nemo-toolkit[all]<=1.20.0,>=1.18.0
rouge_score~=0.1.2
transformers_stream_generator==0.0.4
tiktoken
mpmath==1.3.0