voidful commited on
Commit
549e8a0
1 Parent(s): d87e830

update readme

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -28,7 +28,7 @@ from transformers import AutoTokenizer, AlbertForMaskedLM
28
  import torch
29
  from torch.nn.functional import softmax
30
 
31
- pretrained = './albert_chinese_base'
32
  tokenizer = AutoTokenizer.from_pretrained(pretrained)
33
  model = AlbertForMaskedLM.from_pretrained(pretrained)
34
 
28
  import torch
29
  from torch.nn.functional import softmax
30
 
31
+ pretrained = 'voidful/albert_chinese_base'
32
  tokenizer = AutoTokenizer.from_pretrained(pretrained)
33
  model = AlbertForMaskedLM.from_pretrained(pretrained)
34