llm-trainer / requirements.txt
prasenjeet099's picture
Update requirements.txt
8b32d1a verified
raw
history blame contribute delete
281 Bytes
fastapi==0.110.0
uvicorn==0.27.1
torch==2.2.0
torchvision==0.17.0
numpy==1.26.4
pandas==2.2.0
scikit-learn
matplotlib==3.8.2
huggingface_hub==0.21.3
transformers==4.38.1
datasets==2.18.0
accelerate==0.28.0
django==5.0.2
djangorestframework==3.15.0
requests==2.31.0
python-multipart