--extra-index-url https://pypi.nvidia.com
--extra-index-url https://download.pytorch.org/whl/cu124
accelerate>=0.25.0
build
colored
cuda-python==12.5.0
diffusers>=0.27.0
numpy<2
onnx>=1.12.0
onnx_graphsurgeon>=0.5.2
polygraphy==0.49.9
psutil
pynvml>=11.5.0
pulp
pandas
h5py==3.10.0
pywin32
StrEnum
sentencepiece>=0.1.99
tensorrt~=10.4.0
tokenizers>=0.14
# Default torch is CPU-only on Windows, so need to specify a torch version with GPU support
torch==2.4.1+cu124
torchvision==0.19.1+cu124
nvidia-modelopt[torch]~=0.17.0
transformers>=4.38.2,<=4.45.1
wheel
optimum
evaluate
janus
mpmath==1.3.0
