transformers==4.46.3
einops
addict 
easydict
matplotlib
accelerate