requirements.txt

# Requirements for Head Node
fastapi==0.104.1
uvicorn==0.24.0
requests==2.31.0
pydantic==2.5.0
python-multipart==0.0.6
huggingface_hub==0.20.1
tokenizers==0.15.0
transformers==4.35.2
numpy==1.24.3
pytz==2023.3.post1
aiohttp==3.9.0