oist commited on
Commit
87913b4
·
verified ·
1 Parent(s): 6dc5436

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -278,8 +278,8 @@ mt_embs = encode_mean_pool(mt_sentences, tokenizer, encoder, lang="eng_Latn")
278
  ref_embs = encode_mean_pool(ref_sentences, tokenizer, encoder, lang="eng_Latn")
279
 
280
  # 2. Load BLASER Ref model (ported)
281
- ref_model_name = "oist/blaser-2.0-ref-ported"
282
- ref_model = AutoModel.from_pretrained(qe_model_name, trust_remote_code=True)
283
  ref_model.eval() # set to evaluation mode
284
 
285
  # 3. Compute Ref scores
 
278
  ref_embs = encode_mean_pool(ref_sentences, tokenizer, encoder, lang="eng_Latn")
279
 
280
  # 2. Load BLASER Ref model (ported)
281
+ ref_model_name = "oist/blaser_2_0_ref_ported"
282
+ ref_model = AutoModel.from_pretrained(ref_model_name, trust_remote_code=True)
283
  ref_model.eval() # set to evaluation mode
284
 
285
  # 3. Compute Ref scores