update setup

FlagOpen · Oct 31, 2024 · dd7d32b · dd7d32b
1 parent 6a478a0
commit dd7d32b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name='FlagEmbedding',
-    version='1.3.1',
+    version='1.3.2',
     description='FlagEmbedding',
     long_description=readme,
     long_description_content_type="text/markdown",
@@ -24,4 +24,7 @@
         'sentencepiece',
         'protobuf'
     ],
+    extras_require={
+        'finetune': ['deepspeed', 'flash-attn'],
+    },
 )