suolyer commited on
Commit
33f5a54
1 Parent(s): 73dd8a4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -19,12 +19,12 @@ We collect 4 NLI(Natural Language Inference) datasets in the Chinese domain
19
 
20
  ## Usage
21
  ```python
22
- from transformers import BertForSequenceClassification
23
  from transformers import BertTokenizer
24
  import torch
25
 
26
  tokenizer=BertTokenizer.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-NLI')
27
- model=BertForSequenceClassification.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-NLI')
28
 
29
  texta='今天的饭不好吃'
30
  textb='今天心情不好'
 
19
 
20
  ## Usage
21
  ```python
22
+ from transformers import AutoModelForSequenceClassification
23
  from transformers import BertTokenizer
24
  import torch
25
 
26
  tokenizer=BertTokenizer.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-NLI')
27
+ model=AutoModelForSequenceClassification.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-NLI')
28
 
29
  texta='今天的饭不好吃'
30
  textb='今天心情不好'