deepspeedai · Jul 21, 2023
diff --git a/‎megatron/core/requirements.txt
+3-1 b/‎megatron/core/requirements.txt
+3-1
diff --git a/‎setup.py
+1-1 b/‎setup.py
+1-1
diff --git a/‎tools/__init__.py b/‎tools/__init__.py
diff --git a/‎tools/retro/__init__.py b/‎tools/retro/__init__.py
@@ -1 +1,3 @@
-torch
+pybind11
+torch
+regex
@@ -101,7 +101,7 @@ def req_file(filename, folder="megatron/core"):
         'Natural Language :: English',
         'Operating System :: OS Independent',
     ],
-    packages=['megatron.core', 'megatron.core.pipeline_parallel', 'megatron.core.tensor_parallel'], 
+    packages=setuptools.find_packages(),
     install_requires=install_requires,
 
     # Add in any packaged data.
-Original file line number
+Diff line change
 -torch
 +pybind11
 +torch
 +regex