TensorRT-LLMs/examples/phi/requirements.txt
Kaiyu Xie 035b99e0d0
Update TensorRT-LLM (#1427)
* Update TensorRT-LLM

---------

Co-authored-by: meghagarwal <16129366+megha95@users.noreply.github.com>
2024-04-09 17:03:34 +08:00

7 lines
142 B
Plaintext

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