mirror of
https://github.com/FlagOpen/FlagEmbedding.git
synced 2025-06-27 02:39:58 +00:00
31 lines
803 B
Python
31 lines
803 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.3.5',
|
|
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.44.2',
|
|
'datasets>=2.19.0',
|
|
'accelerate>=0.20.1',
|
|
'sentence_transformers',
|
|
'peft',
|
|
'ir-datasets',
|
|
'sentencepiece',
|
|
'protobuf'
|
|
],
|
|
extras_require={
|
|
'finetune': ['deepspeed', 'flash-attn'],
|
|
},
|
|
)
|