requirements.txt

transformers[torch]>=4.28.1,<5
torch>=1.13.1,<2
accelerate>=0.16.0,<1