Update README.md
Browse files
README.md
CHANGED
|
@@ -15,7 +15,7 @@ Our **JpharmaBERT (base)** is a continually pre-trained version of the BERT mode
|
|
| 15 |
import torch
|
| 16 |
from transformers import AutoModelForMaskedLM, AutoTokenizer, pipeline
|
| 17 |
|
| 18 |
-
model = AutoModelForMaskedLM.from_pretrained("EQUES/jpharma-bert-
|
| 19 |
tokenizer = AutoTokenizer.from_pretrained("EQUES/jpharma-bert-base")
|
| 20 |
fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
|
| 21 |
|
|
|
|
| 15 |
import torch
|
| 16 |
from transformers import AutoModelForMaskedLM, AutoTokenizer, pipeline
|
| 17 |
|
| 18 |
+
model = AutoModelForMaskedLM.from_pretrained("EQUES/jpharma-bert-base", torch_dtype=torch.bfloat16)
|
| 19 |
tokenizer = AutoTokenizer.from_pretrained("EQUES/jpharma-bert-base")
|
| 20 |
fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
|
| 21 |
|