import sys
from setuptools import setup, find_packages

if sys.version_info < (3,):
    raise Exception("Python 2 is not supported by MegatronAdaptor.")

__package_name__ = 'megatron_adaptor'
__version__ = '0.1.0'
__description__ = 'Megatron-LM adaptor for NPU'
__author__ = 'Ascend'
__url__ = 'https://gitcode.com/Ascend/MegatronAdaptor'


setup(
    name=__package_name__,
    version=__version__,
    description=__description__,
    author=__author__,
    url=__url__,
    packages=find_packages(),
    py_modules=['megatron_adaptor'],
    python_requires='>=3.9',
    install_requires=[
        'torch>=2.7.1',
    ],
)