torch>=2.9.1
torchaudio>=2.9.1
timm>=0.9
einops>=0.7
transformers>=4.57,<6