TensorRT-LLMs/examples/quantization/install_requirements.sh
Kaiyu Xie bf0a5afc92
Update TensorRT-LLM (#1598)
* Update TensorRT-LLM
2024-05-14 16:43:41 +08:00

5 lines
238 B
Bash
Executable File

#!/bin/bash
pip3 install nemo_toolkit[all]@git+https://github.com/NVIDIA/NeMo.git@5b5a4445f3de30786d7c2ef4108fc89242a643f3
pip3 install megatron_core@git+https://github.com/NVIDIA/Megatron-LM.git@a5415fcfacef2a37416259bd38b7c4b673583675