Update README.md
Browse files
README.md
CHANGED
|
@@ -70,6 +70,13 @@ This model is a downstream task of [Qwen/Qwen2-1.5B](https://huggingface.co/Qwen
|
|
| 70 |
We leverage the work of [FlagEmbedding reranker](https://github.com/FlagOpen/FlagEmbedding/tree/master/FlagEmbedding/reranker) ,
|
| 71 |
and implement with Qwen2-1.5B as pretrained model.
|
| 72 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
## Usage
|
| 74 |
|
| 75 |
```python
|
|
|
|
| 70 |
We leverage the work of [FlagEmbedding reranker](https://github.com/FlagOpen/FlagEmbedding/tree/master/FlagEmbedding/reranker) ,
|
| 71 |
and implement with Qwen2-1.5B as pretrained model.
|
| 72 |
|
| 73 |
+
## Dependencies
|
| 74 |
+
|
| 75 |
+
```text
|
| 76 |
+
transformers==4.41.2
|
| 77 |
+
flash-attn==2.5.7
|
| 78 |
+
```
|
| 79 |
+
|
| 80 |
## Usage
|
| 81 |
|
| 82 |
```python
|