File size: 1,167 Bytes
b02e3a6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
from setuptools import setup, find_packages
package_name = "llm2vec"
version = {}
with open(f"{package_name}/version.py") as fp:
exec(fp.read(), version)
with open("README.md") as fp:
long_description = fp.read()
setup(
name="llm2vec",
version=version["__version__"],
author="McGill NLP",
author_email=f"parishad.behnamghader@mila.quebec",
url=f"https://github.com/McGill-NLP/{package_name}",
description=f"The official {package_name} library",
python_requires=">=3.8",
packages=find_packages(include=[f"{package_name}*"]),
install_requires=[
"numpy",
"tqdm",
"torch",
"peft",
"transformers>=4.43.1,<=4.44.2",
"datasets",
"evaluate",
"scikit-learn",
],
extras_require={
"evaluation": ["mteb>=1.14.12"],
},
classifiers=[
"Programming Language :: Python :: 3",
"Operating System :: OS Independent",
"License :: OSI Approved :: MIT License",
],
license="MIT",
long_description=long_description,
long_description_content_type="text/markdown",
include_package_data=True,
zip_safe=False,
)
|