torch==2.5.1
transformers==4.40.2
einops==0.8.0