torch
transformers
accelerate
python-slugify
python-dotenv
einops
timm
autotrain-advanced