--extra-index-url https://pypi.ngc.nvidia.com
timm==0.9.16
diffusers==0.21.2
peft==0.10.0
#protobuf==3.19.0
protobuf
#torchvision==0.17.0
transformers==4.37.2
accelerate==0.29.3
loguru==0.7.2
einops==0.7.0
sentencepiece==0.1.99
#cuda-python==11.7.1
onnxruntime==1.17.0
onnx==1.16.0
#onnxruntime==1.12.1
#onnx==1.12.0
#nvidia-pyindex==1.0.9
onnx-graphsurgeon==0.3.27
polygraphy==0.47.1
pandas==2.0.3
gradio==4.31.0
#torch==2.2.0