TensorRT-LLMs/examples/quantization/install_requirements.sh
石晓伟 2a115dae84
Update TensorRT-LLM (#1793)
Co-authored-by: DreamGenX <x@dreamgen.com>
Co-authored-by: Ace-RR <78812427+Ace-RR@users.noreply.github.com>
Co-authored-by: bprus <39293131+bprus@users.noreply.github.com>
Co-authored-by: janpetrov <janpetrov@icloud.com>
2024-06-18 18:18:23 +08:00

5 lines
239 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