diff --git a/requirements-ci.txt b/requirements-ci.txt index f62c18cd4..1e2086a28 100644 --- a/requirements-ci.txt +++ b/requirements-ci.txt @@ -23,4 +23,5 @@ multi_quantization onnx onnxruntime +nvidia-pyindex onnx_graphsurgeon diff --git a/requirements.txt b/requirements.txt index 05b9bfa50..9e8ec9907 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,4 +6,5 @@ typeguard multi_quantization onnx onnxruntime +nvidia-pyindex onnx_graphsurgeon