torch>=1.13.0 transformers>=4.24.0 optimum>=1.5.0 matplotlib