OpenDelta

1

2
import setuptools
3
import os
4
import os
5

6

7
requires = """torch>=1.8.0
8
transformers>=4.10.0
9
datasets>=1.17.0
10
sentencepiece>=0.1.96
11
tqdm>=4.62.2
12
decorator
13
rich
14
web.py
15
gitpython
16
scipy # need?
17
scikit-learn # need?
18
delta_center_client==0.0.4
19
bigmodelvis
20
"""
21

22
def get_requirements():
23
    ret = [x for x in requires.split("\n") if len(x)>0]
24
    print("requirements:", ret)
25
    return ret
26

27

28

29
# path = os.path.dirname(os.path.abspath(__file__))
30
# requires =  get_requirements(path)
31

32
with open('README.md', 'r') as f:
33
    setuptools.setup(
34
        name = 'opendelta',
35
        version = "0.3.2",
36
        description = "An open source framework for delta learning (parameter efficient learning).",
37
        long_description=open("README.md", "r", encoding="utf-8").read(),
38
        long_description_content_type="text/markdown",
39
        author = '',
40
        author_email = 'shengdinghu@gmail.com',
41
        license="Apache",
42
        url="https://github.com/thunlp/OpenDelta",
43
        keywords = ['PLM', 'Parameter-efficient-Learning', 'AI', 'NLP'],
44
        python_requires=">=3.6.0",
45
        install_requires=get_requirements(),
46
        package_dir={'opendelta':'opendelta'},
47
        package_data= {
48
            'opendelta':["utils/interactive/templates/*.html", 'requirments.txt'],
49
        },
50
        include_package_data=True,
51
        packages=setuptools.find_packages(),
52
        classifiers=[
53
            "Programming Language :: Python :: 3",
54
            "Programming Language :: Python :: 3.8",
55
            "Programming Language :: Python :: 3.9",
56
            "Intended Audience :: Developers",
57
            "Intended Audience :: Education",
58
            "Intended Audience :: Science/Research",
59
            "License :: OSI Approved :: Apache Software License",
60
            "Operating System :: OS Independent",
61
        ]
62
    )
63