FlagEmbedding/setup.py
2023-10-12 10:23:17 +08:00

23 lines
604 B
Python

from setuptools import setup, find_packages
with open("README.md", mode="r", encoding="utf-8") as readme_file:
readme = readme_file.read()
setup(
name='FlagEmbedding',
version='1.1.3',
description='FlagEmbedding',
long_description=readme,
long_description_content_type="text/markdown",
author_email='2906698981@qq.com',
url='https://github.com/FlagOpen/FlagEmbedding',
packages=find_packages(),
install_requires=[
'torch>=1.6.0',
'transformers>=4.18.0',
'datasets',
'accelerate>=0.20.1',
'sentence_transformers',
],
)