| 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'], | |
| }, | |
| ) | |