add model weights
Browse files- README.md +3 -5
- pytorch_model.bin +3 -0
README.md
CHANGED
@@ -30,12 +30,10 @@ pip install git+https://github.com/JunnYu/ChineseBert_pytorch.git
|
|
30 |
import torch
|
31 |
from chinesebert import ChineseBertForMaskedLM, ChineseBertTokenizerFast, ChineseBertConfig
|
32 |
|
33 |
-
|
34 |
-
pretrained_model_name = "ShannonAI/ChineseBERT-base"
|
35 |
|
36 |
-
tokenizer = ChineseBertTokenizerFast.from_pretrained(
|
37 |
-
|
38 |
-
chinese_bert = ChineseBertForMaskedLM.from_pretrained(pretrained_model_name, config=config)
|
39 |
|
40 |
text = "北京是[MASK]国的首都。"
|
41 |
inputs = tokenizer(text, return_tensors="pt")
|
|
|
30 |
import torch
|
31 |
from chinesebert import ChineseBertForMaskedLM, ChineseBertTokenizerFast, ChineseBertConfig
|
32 |
|
33 |
+
pretrained_model_name = "junnyu/ChineseBERT-base"
|
|
|
34 |
|
35 |
+
tokenizer = ChineseBertTokenizerFast.from_pretrained(pretrained_model_name)
|
36 |
+
chinese_bert = ChineseBertForMaskedLM.from_pretrained(pretrained_model_name)
|
|
|
37 |
|
38 |
text = "北京是[MASK]国的首都。"
|
39 |
inputs = tokenizer(text, return_tensors="pt")
|
pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c7c3c9f9582e5ce111eb20dbbebab13d9ac2e8f10020b64a7646e195515c8960
|
3 |
+
size 591867879
|