Update README.md
Browse files
README.md
CHANGED
|
@@ -12,6 +12,7 @@ tags: []
|
|
| 12 |
## Model Details
|
| 13 |
### Code to create models
|
| 14 |
```python
|
|
|
|
| 15 |
from transformers import ModernBertConfig, ModernBertForTokenClassification, AutoTokenizer
|
| 16 |
|
| 17 |
model_id = "answerdotai/ModernBERT-base"
|
|
|
|
| 12 |
## Model Details
|
| 13 |
### Code to create models
|
| 14 |
```python
|
| 15 |
+
import torch
|
| 16 |
from transformers import ModernBertConfig, ModernBertForTokenClassification, AutoTokenizer
|
| 17 |
|
| 18 |
model_id = "answerdotai/ModernBERT-base"
|