Overflow-100B / requirements.txt
CooLLaMACEO's picture
Update requirements.txt
0196236 verified
raw
history blame contribute delete
387 Bytes
# Force NumPy 1.x to prevent the Torch crash
numpy<2.0.0
# CPU-optimized Torch
--extra-index-url https://download.pytorch.org/whl/cpu
torch==2.2.1
torchvision==0.17.1
torchaudio==2.2.1
# Missing Tokenizer Dependencies
protobuf
sentencepiece
tiktoken
# Core Logic
transformers==4.38.0
accelerate>=0.27.0
safetensors>=0.4.0
huggingface_hub>=0.20.0
# API Server
fastapi
uvicorn
pydantic