oist
/

blaser_2_0_qe_ported

Sentence Similarity

Model card Files Files and versions

oist commited on Mar 12

Commit

8f8eb8a

·

verified ·

1 Parent(s): 661ae73

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -276,7 +276,7 @@ src_embs = encode_mean_pool(src_sentences, tokenizer, encoder, lang="fra_Latn")
 mt_embs = encode_mean_pool(mt_sentences, tokenizer, encoder, lang="eng_Latn")
 # 2. Load BLASER QE model (ported)
-qe_model_name = "oist/blaser-2.0-qe-ported"
 qe_model = AutoModel.from_pretrained(qe_model_name, trust_remote_code=True)
 qe_model.eval()  # set to evaluation mode

 mt_embs = encode_mean_pool(mt_sentences, tokenizer, encoder, lang="eng_Latn")
 # 2. Load BLASER QE model (ported)
+qe_model_name = "oist/blaser_2_0_qe_ported"
 qe_model = AutoModel.from_pretrained(qe_model_name, trust_remote_code=True)
 qe_model.eval()  # set to evaluation mode