# optimum
git+https://github.com/huggingface/optimum.git#egg=optimum[onnxruntime-gpu]
# accelerate
git+https://github.com/huggingface/accelerate.git
# transformers
git+https://github.com/huggingface/transformers.git

omegaconf==2.3.0
hydra-core==1.3.2
hydra_colorlog==1.2.0

flatten_dict
matplotlib
py3nvml
seaborn
psutil
pandas
rich