timm==0.9.12
transformers==4.36.2
einops==0.7.0
flash_attn==2.5.6