sxtforreal commited on
Commit
16251f0
·
verified ·
1 Parent(s): 6bf841b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -8,7 +8,7 @@ pipeline_tag: feature-extraction
8
  ---
9
 
10
  # Usage
11
- from transformers import AutoTokenizer
12
  from model import (
13
  BERTContrastiveLearning_simcse,
14
  BERTContrastiveLearning_simcse_w,
@@ -29,4 +29,4 @@ model2 = BERTContrastiveLearning_simcse_w.load_from_checkpoint(ckpt2).eval()
29
  model3 = BERTContrastiveLearning_samp.load_from_checkpoint(ckpt3).eval()
30
  model4 = BERTContrastiveLearning_samp_w.load_from_checkpoint(ckpt4).eval()
31
 
32
- cls, _ = model(input_ids, attention_mask) # embeddings
 
8
  ---
9
 
10
  # Usage
11
+ `from transformers import AutoTokenizer
12
  from model import (
13
  BERTContrastiveLearning_simcse,
14
  BERTContrastiveLearning_simcse_w,
 
29
  model3 = BERTContrastiveLearning_samp.load_from_checkpoint(ckpt3).eval()
30
  model4 = BERTContrastiveLearning_samp_w.load_from_checkpoint(ckpt4).eval()
31
 
32
+ cls, _ = model(input_ids, attention_mask) # embeddings`