diff --git a/requirements_gpu.txt b/requirements_gpu.txt index 2fad68e0..3dca9ef7 100644 --- a/requirements_gpu.txt +++ b/requirements_gpu.txt @@ -8,3 +8,4 @@ nvidia-tensorrt onnx_graphsurgeon polygraphy triton-model-analyzer +pytorch-quantization