TensorRT-LLMs/examples/phi/requirements.txt
Kaiyu Xie 5d8ca2faf7
Update TensorRT-LLM (#1639)
* Update TensorRT-LLM

---------

Co-authored-by: vonjackustc <fga@mail.ustc.edu.cn>
2024-05-21 17:51:02 +08:00

7 lines
143 B
Plaintext

--extra-index-url https://pypi.nvidia.com
tensorrt_llm==0.11.0.dev2024052100
datasets~=2.14.5
evaluate~=0.4.1
rouge_score~=0.1.2
einops~=0.7.0