diff --git a/s3_management/manage.py b/s3_management/manage.py
index c6d7d535a..1ba29df45 100644
--- a/s3_management/manage.py
+++ b/s3_management/manage.py
@@ -39,6 +39,7 @@
     "certifi",
     "charset_normalizer",
     "cmake",
+    "colorama",
     "filelock",
     "idna",
     "Jinja2",
@@ -62,6 +63,7 @@
     "torchrec",
     "torchtext",
     "torchvision",
+    "triton",
     "tqdm",
     "typing_extensions",
     "urllib3",