vllm/requirements-neuron.txt
Liangfu Chen 628ec6c17b
[Docker] bump up neuron sdk v2.21 (#11593)
Signed-off-by: Liangfu Chen <liangfc@amazon.com>
2024-12-30 13:46:14 +08:00

8 lines
149 B
Plaintext

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