torch
transformers
diffusers
accelerate
sentencepiece
numpy
tqdm
colorama
gitpython
cupy-cuda12x>=12.0.0
pycuda>=2023.1
nvidia-ml-py3>=7.352.0
psutil>=5.9.0
