Skip to content

Commit dd7d32b

Browse files
committed
update setup
1 parent 6a478a0 commit dd7d32b

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

setup.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
setup(
77
name='FlagEmbedding',
8-
version='1.3.1',
8+
version='1.3.2',
99
description='FlagEmbedding',
1010
long_description=readme,
1111
long_description_content_type="text/markdown",
@@ -24,4 +24,7 @@
2424
'sentencepiece',
2525
'protobuf'
2626
],
27+
extras_require={
28+
'finetune': ['deepspeed', 'flash-attn'],
29+
},
2730
)

0 commit comments

Comments
 (0)