requirements.txt

accelerate==0.18.0
transformers @ git+https://github.com/huggingface/transformers.git@151425ddb29d4ad1a121e8cce62000a2ac52d3ba