gxy commited on
Commit
7a9f37f
1 Parent(s): bd9fee0

fix readme

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -26,7 +26,7 @@ Erlangshen-Deberta-97M-Chinese is pre-trained by bert like mask task from Debert
26
  from transformers import AutoModelForMaskedLM, AutoTokenizer, FillMaskPipeline
27
  import torch
28
 
29
- tokenizer=AutoTokenizer.from_pretrained('IDEA-CCNL/Erlangshen-DeBERTa-v2-97M-Chinese', use_fast=false)
30
  model=AutoModelForMaskedLM.from_pretrained('IDEA-CCNL/Erlangshen-DeBERTa-v2-97M-Chinese')
31
  text = '生活的真谛是[MASK]。'
32
  fillmask_pipe = FillMaskPipeline(model, tokenizer, device=7)
 
26
  from transformers import AutoModelForMaskedLM, AutoTokenizer, FillMaskPipeline
27
  import torch
28
 
29
+ tokenizer=AutoTokenizer.from_pretrained('IDEA-CCNL/Erlangshen-DeBERTa-v2-97M-Chinese', use_fast=False)
30
  model=AutoModelForMaskedLM.from_pretrained('IDEA-CCNL/Erlangshen-DeBERTa-v2-97M-Chinese')
31
  text = '生活的真谛是[MASK]。'
32
  fillmask_pipe = FillMaskPipeline(model, tokenizer, device=7)