torch>=2.6,<3
numpy>=1.26,<3
transformers==4.44.2