gpu_requirements.txt

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

# hydra
omegaconf==2.3.0
hydra-core==1.3.2
hydra_colorlog==1.2.0
hydra-joblib-launcher==1.2.0

# system
psutil==5.9.0
py3nvml==0.2.7

# reporting
flatten_dict
matplotlib
seaborn
pandas
rich