forked from FlagOpen/FlagEmbedding
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
30 lines (28 loc) · 803 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
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.3.2',
description='FlagEmbedding',
long_description=readme,
long_description_content_type="text/markdown",
author_email='[email protected]',
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'],
},
)