requirements.txt

accelerate
timm
einops
torch
torchvision
git+https://github.com/huggingface/transformers.git
sentencepiece
python-dotenv
datasets
sentence-transformers
scikit-learn
numpy