Open-Llama/requirements.txt
2023-04-29 13:39:03 +08:00

20 lines
256 B
Plaintext

torch==1.13.1
torchvision
torchaudio
zstandard
accelerate
datasets
wandb
deepspeed
absl-py
torchinfo
scikit-learn
datasets==2.10.1
matplotlib
seaborn
sentencepiece
triton
functorch==1.13.1
xformers
gradio
git+https://github.com/huggingface/transformers.git