BM-K
/

KoSimCSE-bert

Feature Extraction

text-embeddings-inference

Model card Files Files and versions

BM-K commited on May 23, 2022

Commit

3c0ed9c

·

1 Parent(s): 5e6ebe7

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -16,7 +16,8 @@ def cal_score(a, b):
     b_norm = b / b.norm(dim=1)[:, None]
     return torch.mm(a_norm, b_norm.transpose(0, 1)) * 100
-model = AutoModel.from_pretrained('BM-K/KoSimCSE-bert') AutoTokenizer.from_pretrained('BM-K/KoSimCSE-bert')
 sentences = ['치타가 들판을 가로 질러 먹이를 쫓는다.',
              '치타 한 마리가 먹이 뒤에서 달리고 있다.',

     b_norm = b / b.norm(dim=1)[:, None]
     return torch.mm(a_norm, b_norm.transpose(0, 1)) * 100
+model = AutoModel.from_pretrained('BM-K/KoSimCSE-bert')
+AutoTokenizer.from_pretrained('BM-K/KoSimCSE-bert')
 sentences = ['치타가 들판을 가로 질러 먹이를 쫓는다.',
              '치타 한 마리가 먹이 뒤에서 달리고 있다.',