mirror of
https://github.com/FlagOpen/FlagEmbedding.git
synced 2025-06-27 02:39:58 +00:00
30 lines
776 B
Python
30 lines
776 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.0',
|
|
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',
|
|
'deepspeed',
|
|
'flash-attn==2.5.6',
|
|
'ir-datasets',
|
|
'sentencepiece',
|
|
'protobuf'
|
|
],
|
|
)
|