numpy<2
transformers==4.36.0
torch==2.1.0
fastapi
uvicorn
safetensors
accelerate