torch==2.1.0 accelerate==0.26.1 transformers==4.38.0 optimum[onnxruntime]==1.17.1