trainSpace / requirements.txt
pathii's picture
Update requirements.txt
4b7072d verified
raw
history blame contribute delete
368 Bytes
# Core packages
torch==2.2.1 ; sys_platform != 'win32'
transformers==4.38.2
datasets==2.18.0
accelerate==0.27.2
# Optimization packages
deepspeed==0.13.1 ; sys_platform != 'win32'
vllm==0.3.0 ; sys_platform != 'win32'
unsloth==0.2.0 ; sys_platform != 'win32'
# API packages
fastapi==0.109.1
uvicorn==0.27.0
# Build dependencies
ninja==1.11.1
huggingface_hub==0.20.3