torch 
transformers~=4.6.0
scipy