apex==0.9.10dev
accelerate
colored
controlnet_aux==0.0.6
cuda-python
# TODO: Pin Diffusers version after the next release
git+https://github.com/huggingface/diffusers.git@3335e2262d47e7d7e311a44dea7f454b5f01b643 # Install from source for the latest changes in main
ftfy
matplotlib
nvtx
onnx==1.17.0
onnxruntime==1.19.2
opencv-python-headless==4.8.0.74
scipy
transformers==4.42.2
--extra-index-url https://pypi.nvidia.com
nvidia-modelopt[torch,onnx]==0.21.0
onnx-graphsurgeon==0.5.2
peft==0.13.0
polygraphy==0.49.9
sentencepiece
