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-04 15:24:39 +08:00
|
|
|
name='flag_embedding',
|
2023-08-05 20:17:24 +08:00
|
|
|
version='1.0.1',
|
|
|
|
|
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-04 15:24:39 +08:00
|
|
|
package_dir={"": "flag_embedding"},
|
|
|
|
|
packages=find_packages("flag_embedding"),
|
2023-08-02 17:40:00 +08:00
|
|
|
install_requires=[
|
|
|
|
|
'torch>=1.6.0',
|
|
|
|
|
'transformers>=4.18.0',
|
|
|
|
|
'datasets',
|
2023-08-03 10:55:42 +08:00
|
|
|
'accelerate>=0.20.1'
|
2023-08-02 17:40:00 +08:00
|
|
|
],
|
|
|
|
|
)
|
