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",