From 0903e78e8ad93ee64382aa5250cb3eba0559bbf3 Mon Sep 17 00:00:00 2001 From: Omkar Salpekar Date: Mon, 11 Sep 2023 13:46:24 -0700 Subject: [PATCH] Add torch-tensorrt to S3 PyPI Index As pytorch/tensorrt moves off of CCI onto Nova, we must to host their nightlies on our S3 index. This change allows the indexing to occur correctly for this package. --- s3_management/manage.py | 1 + 1 file changed, 1 insertion(+) diff --git a/s3_management/manage.py b/s3_management/manage.py index 4b35ecab2..719923d47 100644 --- a/s3_management/manage.py +++ b/s3_management/manage.py @@ -80,6 +80,7 @@ "requests", "sympy", "torch", + "torch_tensorrt", "torcharrow", "torchaudio", "torchcsprng",