KordAI
/

III-BERT

masked-language-model

pretrained-model

custom-transformer

Model card Files Files and versions

Naphon commited on 16 days ago

Commit

2e9bcac

·

verified ·

1 Parent(s): 3c14780

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -44,11 +44,11 @@ The repository contains custom model code. Review that code before loading it an
 ```python
 import torch
-from transformers import AutoTokenizer, AutoModelForMaskedLM
 model_id = "KordAI/III-BERT"
-tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
 model = AutoModelForMaskedLM.from_pretrained(
     model_id,
     trust_remote_code=True,

 ```python
 import torch
+from transformers import CamembertTokenizer, AutoModelForMaskedLM
 model_id = "KordAI/III-BERT"
+tokenizer = CamembertTokenizer.from_pretrained(model_id, trust_remote_code=True)
 model = AutoModelForMaskedLM.from_pretrained(
     model_id,
     trust_remote_code=True,