diff --git a/s3_management/manage.py b/s3_management/manage.py
index f6d64b928c..0dbc6fb067 100755
--- a/s3_management/manage.py
+++ b/s3_management/manage.py
@@ -109,6 +109,7 @@
     "filelock",
     "fsspec",
     "idna",
+    "iopath",
     "intel_openmp",
     "Jinja2",
     "lit",
@@ -143,6 +144,7 @@
     "nvidia_nvjitlink_cu12",
     "packaging",
     "portalocker",
+    "pyre_extensions",
     "pytorch_triton",
     "pytorch_triton_rocm",
     "pytorch_triton_xpu",