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