Update README.md
Browse files
README.md
CHANGED
|
@@ -40,6 +40,19 @@ pipeline_tag: text-ranking
|
|
| 40 |
|
| 41 |
## 使い方
|
| 42 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
### SentenceTransformers
|
| 44 |
|
| 45 |
```python
|
|
|
|
| 40 |
|
| 41 |
## 使い方
|
| 42 |
|
| 43 |
+
動作には transformers ライブラリの v4.48 以上が必要です。
|
| 44 |
+
|
| 45 |
+
```
|
| 46 |
+
pip install -U "transformers>=4.48.0" sentence-transformers
|
| 47 |
+
```
|
| 48 |
+
|
| 49 |
+
GPU が Flash Attention 2 をサポートしている場合、flash-attn ライブラリを入れることで、高速な推論が可能です。
|
| 50 |
+
|
| 51 |
+
```
|
| 52 |
+
pip install flash-attn --no-build-isolation
|
| 53 |
+
```
|
| 54 |
+
|
| 55 |
+
|
| 56 |
### SentenceTransformers
|
| 57 |
|
| 58 |
```python
|