TensorRT-LLMs/examples/mamba/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

7 lines
94 B
Plaintext

tensorrt_llm==0.15.0
transformers>=4.39.0
datasets~=2.14.5
evaluate
rouge_score
sentencepiece