termcolor
matplotlib
pyyaml
fire
transformers>=4.34.0
matplotlib
gpustat
