vllm/requirements-neuron.txt
omrishiv 7f9c8902e3
[Hardware][AWS] update neuron to 2.20 (#8676)
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com>
2024-09-20 15:19:44 -07:00

8 lines
149 B
Plaintext

# Common dependencies
-r requirements-common.txt
# Dependencies for Neuron devices
transformers-neuronx >= 0.12.0
torch-neuronx >= 2.1.2
neuronx-cc