1import torch
2from sentence_transformers import SentenceTransformer
3from sentence_transformers.util import cos_sim
4
5model1 = SentenceTransformer(
6 "perplexity-ai/pplx-embed-v1-4B",
7 trust_remote_code=True,
8 model_kwargs={"dtype": torch.float16}
9)
10model2 = SentenceTransformer(
11 "mykor/pplx-embed-v1-4b",
12 model_kwargs={"dtype": torch.float16}
13)
14
15text = """사라져가네 멀어져 가네
16흩어지듯이
17잊혀져가네 어두워지네
18더 못 볼 것처럼
19사라져 가는 멀어져 가는
20널 따라가봐도
21여전히 그 소린 멈춰있어
22다 지나도
23더 바래도
24달빛처럼 비추는 넌
25아름다워줘
26다시 밝게 빛나줘
27네가 있어 여긴 여전하게
28살아있어
29그대로 있어줘
30그대로 불러줘
31아직도
32내겐 잊지 못할 기억이라
33
34지워내봐도 희미해져도
35가릴 수 없이
36작아져버린 흘러가버린
37그때의 목소리가
38기억조차 못 했을 거리
39마저 내 눈앞에 그려서
40살아져가게 해
41살아져가게 날
42이대로
43꼭 너만은
44아름다워줘
45다시 밝게 빛나줘
46네가 있어 여긴 여전하게
47살아있어
48그대로 있어줘
49그대로 불러줘
50아직도
51내겐 잊지 못할 기억이라
52
53순간을 그려낸
54시간이
55세상에 번져 있어
56아름다웠던
57그리워했던
58시간 속에 우린
59영원하게 멈춰있어
60그대로 웃어줘
61그대로 믿어줘
62아직도
63내겐 잊지 못할 기억이라
64기억이라
65기억이라
66너는 이 순간의 기적이라"""
67
68embed1 = model1.encode(text)
69embed2 = model2.encode(text)
70print(cos_sim(embed1, embed2).item())
