torch
optimum
transformers>=4.26
transformers-cfg==0.2.0