Update README.md
Browse files
README.md
CHANGED
|
@@ -11,7 +11,7 @@ Currently Adding Token Classification to mistral model ;
|
|
| 11 |
Still need some kinks sorting out -on the way - Testing
|
| 12 |
|
| 13 |
```
|
| 14 |
-
|
| 15 |
from transformers import AutoModelForTokenClassification, AutoTokenizer
|
| 16 |
import torch
|
| 17 |
|
|
|
|
| 11 |
Still need some kinks sorting out -on the way - Testing
|
| 12 |
|
| 13 |
```
|
| 14 |
+
! pip install flash_attn
|
| 15 |
from transformers import AutoModelForTokenClassification, AutoTokenizer
|
| 16 |
import torch
|
| 17 |
|