Update README.md
Browse files
README.md
CHANGED
|
@@ -49,7 +49,7 @@ nDCG and Recall scores of this model(out-of-domain predictions) and other multil
|
|
| 49 |
alt="drawing"/>
|
| 50 |
|
| 51 |
## Usage
|
| 52 |
-
First install required libraries (Flash Attention is a must for consistency):
|
| 53 |
|
| 54 |
```bash
|
| 55 |
pip install -U einops flash_attn
|
|
|
|
| 49 |
alt="drawing"/>
|
| 50 |
|
| 51 |
## Usage
|
| 52 |
+
First install required libraries (Flash Attention supporting GPU is a must for consistency):
|
| 53 |
|
| 54 |
```bash
|
| 55 |
pip install -U einops flash_attn
|