environment:
  - HF_TRUST_REMOTE_CODE=true
python_packages:
  - flash-attn
  - timm
  - einops