Commit c60fb37e authored by Evelina Bakhturina's avatar Evelina Bakhturina
Browse files

update 



Signed-off-by: default avatarEvelina Bakhturina <ebakhturina@nvidia.com>
parent c15829b5

megatron/__init__.py

+1 −0
Original line number Diff line number Diff line
@@ -15,6 +15,7 @@ 


from .package_info import (

    __description__,

    __contact_names__,

    __url__,

    __download_url__,

    __keywords__,

megatron/package_info.py

+2 −1
Original line number Diff line number Diff line
@@ -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

+2 −0
Original line number Diff line number Diff line
@@ -3,3 +3,5 @@ torch 
torchvision

six

regex

numpy

setup.py

+3 −0
Original line number Diff line number Diff line
@@ -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=[