Update README.md
Browse files
README.md
CHANGED
|
@@ -67,7 +67,6 @@ pip install mxbai-rerank
|
|
| 67 |
```python
|
| 68 |
from mxbai_rerank import MxbaiRerankV2
|
| 69 |
|
| 70 |
-
# pass attn_implementation="flash_attention_2" to use flash attention
|
| 71 |
model = MxbaiRerankV2("mixedbread-ai/mxbai-rerank-large-v2")
|
| 72 |
|
| 73 |
query = "Who wrote 'To Kill a Mockingbird'?"
|
|
|
|
| 67 |
```python
|
| 68 |
from mxbai_rerank import MxbaiRerankV2
|
| 69 |
|
|
|
|
| 70 |
model = MxbaiRerankV2("mixedbread-ai/mxbai-rerank-large-v2")
|
| 71 |
|
| 72 |
query = "Who wrote 'To Kill a Mockingbird'?"
|