1
0

requirements-neuron.txt 244 B

123456789101112
  1. sentencepiece # Required for LLaMA tokenizer.
  2. numpy
  3. transformers-neuronx >= 0.9.0
  4. torch-neuronx >= 2.1.0
  5. neuronx-cc
  6. fastapi
  7. uvicorn[standard]
  8. pydantic >= 2.0 # Required for OpenAI server.
  9. prometheus_client >= 0.18.0
  10. requests
  11. psutil
  12. py-cpuinfo