torch
torchvision
torchaudio
git+https://github.com/zphang/transformers@llama_push
sentencepiece
accelerate
datasets
torchmetrics
evaluate
transformers
wandb
peft
nodelist-inflator
deepspeed
jsonlines