requirements.txt

accelerate==0.25.0
huggingface-hub==0.20.1
torch==2.1.2
torchvision==0.16.2
transformers==4.36.2
einops==0.7.0