mirror of
https://github.com/FlagOpen/FlagEmbedding.git
synced 2025-12-30 08:42:24 +00:00
30 lines
772 B
Python
30 lines
772 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.2.11',
|
|
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(),
|
|
include_package_data=True,
|
|
install_requires=[
|
|
'torch>=1.6.0',
|
|
'transformers>=4.33.0',
|
|
'datasets>=2.19.0',
|
|
'accelerate>=0.20.1',
|
|
'sentence_transformers',
|
|
'peft',
|
|
'beir',
|
|
'deepspeed',
|
|
'flash-attn==2.5.6',
|
|
'mteb==1.15.0',
|
|
'ir-datasets'
|
|
],
|
|
)
|