OpenDeltaMirror/setup.py

60 lines
1.7 KiB
Python
Raw Normal View History

2022-02-14 21:19:03 +08:00
import setuptools
import os
2022-06-07 01:52:32 +08:00
import os
2022-02-14 21:19:03 +08:00
2022-07-06 22:00:58 +08:00
requires = """torch>=1.8.0
transformers>=4.10.0
datasets==1.17.0
sentencepiece>=0.1.96
tqdm>=4.62.2
# loralib
decorator
rich
web.py
gitpython
2022-07-08 11:48:02 +08:00
delta_center_client==0.0.4
2022-07-06 22:00:58 +08:00
"""
def get_requirements():
ret = [x for x in requires.split("\n") if len(x)>0]
print("requirements:", ret)
2022-02-14 21:19:03 +08:00
return ret
2022-07-06 22:00:58 +08:00
# path = os.path.dirname(os.path.abspath(__file__))
# requires = get_requirements(path)
2022-02-14 21:19:03 +08:00
with open('README.md', 'r') as f:
setuptools.setup(
name = 'opendelta',
2022-07-08 11:48:02 +08:00
version = "0.2.4",
2022-02-14 21:19:03 +08:00
description = "An open source framework for delta learning (parameter efficient learning).",
long_description=open("README.md", "r", encoding="utf-8").read(),
long_description_content_type="text/markdown",
author = '',
author_email = 'shengdinghu@gmail.com',
license="Apache",
url="https://github.com/thunlp/OpenDelta",
keywords = ['PLM', 'Parameter-efficient-Learning', 'AI', 'NLP'],
2022-03-20 03:09:00 +08:00
python_requires=">=3.6.0",
2022-07-06 22:00:58 +08:00
install_requires=get_requirements(),
2022-06-07 01:52:32 +08:00
package_dir={'opendelta':'opendelta'},
package_data= {
2022-07-03 10:30:59 +08:00
'opendelta':["utils/interactive/templates/*.html", 'requirments.txt'],
2022-06-07 01:52:32 +08:00
},
include_package_data=True,
2022-02-14 21:19:03 +08:00
packages=setuptools.find_packages(),
classifiers=[
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Intended Audience :: Developers",
"Intended Audience :: Education",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: Apache Software License",
"Operating System :: OS Independent",
]
)