torch==2.0.1
torchvision==0.15.2
torchaudio==2.0.2
safetensors==0.3.0
datasets==2.10.1
sentencepiece
git+https://github.com/huggingface/transformers
accelerate==0.17.1
triton==2.0.0
texttable
toml
numpy
protobuf==3.20.2