shibing624
/

macbert4csc-base-chinese

Text Generation

Model card Files Files and versions

shibing624 commited on Jan 29, 2022

Commit

a3383e2

·

1 Parent(s): 1823f8a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -47,7 +47,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 tokenizer = BertTokenizer.from_pretrained("shibing624/macbert4csc-base-chinese")
 model = BertForMaskedLM.from_pretrained("shibing624/macbert4csc-base-chinese")
-model = model.to(device)
 texts = ["今天新情很好", "你找到你最喜欢的工作，我也很高心。"]
 with torch.no_grad():

 tokenizer = BertTokenizer.from_pretrained("shibing624/macbert4csc-base-chinese")
 model = BertForMaskedLM.from_pretrained("shibing624/macbert4csc-base-chinese")
+model.to(device)
 texts = ["今天新情很好", "你找到你最喜欢的工作，我也很高心。"]
 with torch.no_grad():