EQUES
/

jpharma-bert-base

Model card Files Files and versions

tkr12 commited on Jun 15

Commit

9094568

·

verified ·

1 Parent(s): e1cb868

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ Our **JpharmaBERT (base)** is a continually pre-trained version of the BERT mode
 import torch
 from transformers import AutoModelForMaskedLM, AutoTokenizer, pipeline
-model = AutoModelForMaskedLM.from_pretrained("EQUES/jpharma-bert-large", torch_dtype=torch.bfloat16)
 tokenizer = AutoTokenizer.from_pretrained("EQUES/jpharma-bert-base")
 fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)

 import torch
 from transformers import AutoModelForMaskedLM, AutoTokenizer, pipeline
+model = AutoModelForMaskedLM.from_pretrained("EQUES/jpharma-bert-base", torch_dtype=torch.bfloat16)
 tokenizer = AutoTokenizer.from_pretrained("EQUES/jpharma-bert-base")
 fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)