TensorRT-LLMs/examples/mamba/requirements.txt
2025-04-09 09:06:37 +08:00

7 lines
94 B
Plaintext

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