update

Signed-off-by: Evelina Bakhturina <ebakhturina@nvidia.com>

megatron/__init__.py

@@ -15,6 +15,7 @@
 
 from .package_info import (
     __description__,
+    __contact_names__,
     __url__,
     __download_url__,
     __keywords__,

megatron/package_info.py

@@ -17,11 +17,12 @@
 MAJOR = 1
 MINOR = 0
 
-# Use the following formatting: (major, minor, patch)
+# Use the following formatting: (major, minor)
 VERSION = (MAJOR, MINOR)
 
 __version__ = '.'.join(map(str, VERSION))
 __package_name__ = 'megatron-lm'
+__contact_names__ = 'NVIDIA INC'
 __url__ = 'https://github.com/NVIDIA/Megatron-LM'
 __download_url__ = 'https://github.com/NVIDIA/Megatron-LM/releases'
 __description__ = 'Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelism.'

requirements.txt

@@ -3,3 +3,5 @@ torch
 torchvision
 six
 regex
+numpy
+

setup.py

@@ -24,6 +24,7 @@ if sys.version_info < (3,):
 
 from megatron.package_info import (
     __description__,
+    __contact_names__,
     __url__,
     __download_url__,
     __keywords__,
@@ -61,6 +62,8 @@ setuptools.setup(
     long_description_content_type="text/markdown",
     # The project's main homepage.
     url=__url__,
+    author=__contact_names__,
+    maintainer=__contact_names__,
     # The licence under which the project is released
     license=__license__,
     classifiers=[