TensorRT-LLMs/examples/chatglm/requirements.txt
Kaiyu Xie f14d1d433c
Update TensorRT-LLM (#2389)
* Update TensorRT-LLM

---------

Co-authored-by: Alessio Netti <netti.alessio@gmail.com>
2024-10-29 22:24:38 +08:00

11 lines
230 B
Plaintext

--extra-index-url https://pypi.nvidia.com
tensorrt_llm==0.15.0.dev2024102900
datasets~=2.14.5
evaluate~=0.4.1
protobuf
rouge_score~=0.1.2
sentencepiece
tiktoken
# https://github.com/THUDM/ChatGLM3/issues/1324
transformers<=4.43.0