torch
transformers
accelerate
python-slugify
python-dotenv
einops
timm
flash-attn