api / requirements.txt
gary-boon
Pin PyTorch <2.11.0 to fix CUDA driver incompatibility on HF Spaces
8b9ad4b
raw
history blame contribute delete
700 Bytes
# Core dependencies for HuggingFace Spaces deployment
fastapi==0.104.1
uvicorn[standard]==0.24.0
websockets==12.0
python-multipart==0.0.6
pydantic>=2.0.0 # Relaxed for mistral-common compatibility
# Machine Learning
# torch 2.3+ required for transformers 4.44+ (pytree API compatibility)
# Pinned <2.11: PyTorch 2.11.0 requires CUDA drivers newer than HF Spaces GPU provides
torch>=2.3.0,<2.11.0
transformers>=4.44.0
accelerate>=0.30.0
mistral-common>=1.5.0 # Required for Devstral Tekken tokenizer
# Utilities
numpy>=1.24.0,<2.0 # Relaxed for mistral-common compatibility
aiofiles==23.2.1
python-dotenv==1.0.0
zarr==2.14.2
numcodecs==0.11.0
# SWE-bench support
datasets==2.16.1
pyarrow==14.0.2