torch 
transformers==4.41.2