chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,30 @@
|
||||
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.4.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,<6.0.0',
|
||||
'datasets>=2.19.0',
|
||||
'accelerate>=0.20.1',
|
||||
'sentence_transformers',
|
||||
'peft',
|
||||
'ir-datasets',
|
||||
'sentencepiece',
|
||||
'protobuf'
|
||||
],
|
||||
extras_require={
|
||||
'finetune': ['deepspeed', 'flash-attn'],
|
||||
},
|
||||
)
|
||||
Reference in New Issue
Block a user