voidful
/

albert_chinese_base

Inference Endpoints

Model card Files Files and versions Community

voidful commited on Aug 3, 2021

Commit

549e8a0

•

1 Parent(s): d87e830

update readme

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ from transformers import AutoTokenizer, AlbertForMaskedLM
 import torch
 from torch.nn.functional import softmax
-pretrained = './albert_chinese_base'
 tokenizer = AutoTokenizer.from_pretrained(pretrained)
 model = AlbertForMaskedLM.from_pretrained(pretrained)

 import torch
 from torch.nn.functional import softmax
+pretrained = 'voidful/albert_chinese_base'
 tokenizer = AutoTokenizer.from_pretrained(pretrained)
 model = AlbertForMaskedLM.from_pretrained(pretrained)