Update README.md
Browse files
README.md
CHANGED
|
@@ -10,8 +10,8 @@ import torch
|
|
| 10 |
from transformers import AutoTokenizer
|
| 11 |
from modeling_reward import BERTRewardModel
|
| 12 |
|
| 13 |
-
|
| 14 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
| 15 |
model = BERTRewardModel(model_name="answerdotai/ModernBERT-base")
|
| 16 |
state_dict = torch.load("model.safetensors") # or use safetensors.torch.load_file
|
| 17 |
model.load_state_dict(state_dict)
|
|
|
|
| 10 |
from transformers import AutoTokenizer
|
| 11 |
from modeling_reward import BERTRewardModel
|
| 12 |
|
| 13 |
+
model_name = DarianNLP/modernbert-nl-sql
|
| 14 |
+
tokenizer = AutoTokenizer.from_pretrained(DarianNLP/modernbert-nl-sql, trust_remote_code=True)
|
| 15 |
model = BERTRewardModel(model_name="answerdotai/ModernBERT-base")
|
| 16 |
state_dict = torch.load("model.safetensors") # or use safetensors.torch.load_file
|
| 17 |
model.load_state_dict(state_dict)
|