Open-Llama/requirements.txt

21 lines
238 B
Plaintext

torch==1.13.1
torchvision
torchaudio
zstandard
accelerate
wandb
deepspeed
absl-py
torchinfo
scikit-learn
datasets==2.10.1
matplotlib
seaborn
sentencepiece
triton
functorch==1.13.1
xformers==0.0.16
gradio
peft
transformers
fsspec==2023.9.2