llama-65b-hf / requirements.txt
dontito's picture
Update requirements.txt
22b1a9b
torch
torchvision
torchaudio
git+https://github.com/zphang/transformers@llama_push
sentencepiece
accelerate
datasets
torchmetrics
evaluate
transformers
wandb
peft
nodelist-inflator
deepspeed
jsonlines