accelerate @ git+https://github.com/huggingface/accelerate
transformers
timm
huggingface_hub
tabulate
flash-attn
einops