requirements.txt

torch 
git+https://github.com/shumingma/transformers.git
accelerate