Skip to content

Commit b198041

Browse files
committed
mv deepspeed flash-attn to FT
1 parent 390b893 commit b198041

4 files changed

Lines changed: 17 additions & 2 deletions

File tree

examples/README.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,11 @@ print(scores)
6969

7070
We support fine-tuning a variety of BGE series models, including `bge-large-en-v1.5`, `bge-m3`, `bge-en-icl`, `bge-multilingual-gemma2`, `bge-reranker-v2-m3`, `bge-reranker-v2-gemma`, and `bge-reranker-v2-minicpm-layerwise`, among others. As examples, we use the basic models `bge-large-en-v1.5` and `bge-reranker-large`. For more details, please refer to the [embedder](https://github.com/FlagOpen/FlagEmbedding/tree/master/examples/finetune/embedder) and [reranker](https://github.com/FlagOpen/FlagEmbedding/tree/master/examples/finetune/reranker) sections.
7171

72+
```shell
73+
pip install deepspeed
74+
pip install flash-attn --no-build-isolation
75+
```
76+
7277
## 1. Embedder
7378

7479
```shell

examples/finetune/embedder/README.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,8 @@ In this example, we show how to finetune the embedder with your data.
88

99
```shell
1010
pip install -U FlagEmbedding
11+
pip install deepspeed
12+
pip install flash-attn --no-build-isolation
1113
```
1214

1315
- **from source**
@@ -16,12 +18,16 @@ pip install -U FlagEmbedding
1618
git clone https://github.com/FlagOpen/FlagEmbedding.git
1719
cd FlagEmbedding
1820
pip install .
21+
pip install deepspeed
22+
pip install flash-attn --no-build-isolation
1923
```
2024

2125
For development, install as editable:
2226

2327
```shell
2428
pip install -e .
29+
pip install deepspeed
30+
pip install flash-attn --no-build-isolation
2531
```
2632

2733
## 2. Data format

examples/finetune/reranker/README.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,8 @@ In this example, we show how to finetune the reranker with your data.
88

99
```shell
1010
pip install -U FlagEmbedding
11+
pip install deepspeed
12+
pip install flash-attn --no-build-isolation
1113
```
1214

1315
- **from source**
@@ -16,12 +18,16 @@ pip install -U FlagEmbedding
1618
git clone https://github.com/FlagOpen/FlagEmbedding.git
1719
cd FlagEmbedding
1820
pip install .
21+
pip install deepspeed
22+
pip install flash-attn --no-build-isolation
1923
```
2024

2125
For development, install as editable:
2226

2327
```shell
2428
pip install -e .
29+
pip install deepspeed
30+
pip install flash-attn --no-build-isolation
2531
```
2632

2733
## 2. Data format

setup.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,6 @@
2020
'accelerate>=0.20.1',
2121
'sentence_transformers',
2222
'peft',
23-
'deepspeed',
24-
'flash-attn==2.5.6',
2523
'ir-datasets',
2624
'sentencepiece',
2725
'protobuf'

0 commit comments

Comments
 (0)