serpent / requirements.txt
kfoughali's picture
Update requirements.txt
738d9fd verified
raw
history blame contribute delete
353 Bytes
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.5.1
transformers>=4.45.0
accelerate>=0.30.0
sentencepiece>=0.1.99
protobuf>=3.20.0
huggingface_hub>=0.16.0
datasets>=2.14.0
pandas>=1.5.0
tqdm>=4.65.0
scikit-learn>=1.0.0
scipy>=1.8.0
psutil>=5.9.0
pynvml>=11.5.0
matplotlib>=3.5.0
seaborn>=0.11.0
bitsandbytes>=0.41.0
gradio>=4.0.0