TensorRT-LLMs/examples/phi/requirements.txt
Dan Blanaru 48686bca3a
open source 7f370deb0090d885d7518c2b146399ba3933c004 (#2273)
* Update TensorRT-LLM

---------
Co-authored-by: Qingquan Song <ustcsqq@gmail.com>
2024-09-30 13:51:19 +02:00

8 lines
159 B
Plaintext

--extra-index-url https://pypi.nvidia.com
tensorrt_llm==0.14.0.dev2024100100
datasets~=2.14.5
evaluate~=0.4.1
rouge_score~=0.1.2
einops~=0.7.0
tiktoken==0.6.0