torch==2.6.0
transformers >= 4.53.0
einops
safetensors