2023-08-02 17:40:00 +08:00
|
|
|
from setuptools import setup, find_packages
|
|
|
|
|
|
2023-08-05 20:17:24 +08:00
|
|
|
with open("README.md", mode="r", encoding="utf-8") as readme_file:
|
|
|
|
|
readme = readme_file.read()
|
|
|
|
|
|
2023-08-02 17:40:00 +08:00
|
|
|
setup(
|
2023-08-05 22:53:05 +08:00
|
|
|
name='FlagEmbedding',
|
2023-12-14 17:12:14 +08:00
|
|
|
version='1.1.7',
|
2023-08-05 20:17:24 +08:00
|
|
|
description='FlagEmbedding',
|
|
|
|
|
long_description=readme,
|
|
|
|
|
long_description_content_type="text/markdown",
|
2023-08-05 20:06:27 +08:00
|
|
|
author_email='2906698981@qq.com',
|
|
|
|
|
url='https://github.com/FlagOpen/FlagEmbedding',
|
2023-08-05 22:53:05 +08:00
|
|
|
packages=find_packages(),
|
2023-08-02 17:40:00 +08:00
|
|
|
install_requires=[
|
|
|
|
|
'torch>=1.6.0',
|
2023-12-18 19:56:14 +08:00
|
|
|
'transformers==4.34.0',
|
2023-08-02 17:40:00 +08:00
|
|
|
'datasets',
|
2023-09-28 00:49:16 +08:00
|
|
|
'accelerate>=0.20.1',
|
|
|
|
|
'sentence_transformers',
|
2023-08-02 17:40:00 +08:00
|
|
|
],
|
|
|
|
|
)
|
