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