torch==2.4.1
transformers==4.39.0
einops