junnyu commited on
Commit
a25dde7
·
1 Parent(s): 14113ab

add model weights

Browse files
Files changed (2) hide show
  1. README.md +3 -5
  2. 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
- pretrained_tokenizer_name = "junnyu/ChineseBERT-base"
34
- pretrained_model_name = "ShannonAI/ChineseBERT-base"
35
 
36
- tokenizer = ChineseBertTokenizerFast.from_pretrained(pretrained_tokenizer_name)
37
- config = ChineseBertConfig.from_pretrained(pretrained_tokenizer_name)
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