|
# Core dependencies first |
|
packaging>=21.0 |
|
wheel>=0.37.0 |
|
setuptools>=65.0.0 |
|
|
|
# PyTorch and ML libraries |
|
torch>=2.0.0 |
|
transformers>=4.35.0 |
|
peft>=0.5.0 |
|
accelerate>=0.24.0 |
|
tokenizers>=0.15.0 |
|
bitsandbytes>=0.41.0 |
|
scipy>=1.11.0 |
|
numpy>=1.24.0 |
|
sentencepiece>=0.1.99 |
|
protobuf>=3.20.0 |
|
safetensors>=0.3.0 |
|
|
|
# FastAPI and web |
|
fastapi>=0.104.0 |
|
uvicorn[standard]>=0.24.0 |
|
pydantic>=2.0.0 |
|
|
|
# Build tools for Flash Attention |
|
ninja>=1.10.0 |
|
|
|
# Flash Attention (optional - will fallback to eager if fails) |
|
# Remove this line if causing build issues |
|
# flash-attn>=2.3.0 |