Views
No views yet
1import numpy as np
2from llama_cpp import Llama
3from sentence_transformers import SentenceTransformer
4from sentence_transformers.util import cos_sim
5
6model = SentenceTransformer(
7 "ibm-granite/granite-embedding-311m-multilingual-r2",
8)
9llama = Llama.from_pretrained(
10 repo_id="mykor/granite-embedding-311m-multilingual-r2-GGUF",
11 filename="granite-embedding-311M-multilingual-r2-BF16.gguf",
12 verbose=False,
13 embedding=True,
14 n_ctx=0,
15)
16
17text = """이젠 깨어나 설레는 이 마음 따라
18기다리던 오늘을 만들 거야
19
20이 많은 시간을 난
21흘리고 싶지는 않아
22
23이제 나는 분명히
24준비가 돼 있는데
25
26어떤 무기력함이 와도
27나에겐 의미가 없을 만큼
28
29또다시 wake up 더 wake up
30달궈져 더 뜨겁게
31
32굳이 봐줄 필요 없어 어차피 never give up
33뭐가 덤벼오든 나는 어쩌라 하듯이
34뭐든 할 수 있을 거 같아
35
36이젠 깨어나 설레는 이 마음 따라
37기다리던 오늘을 만들 거야
38
39도시 사이에 피어나 이겨내온 꽃처럼
40분명 피워낼 거야 나를
41
42나에게 남아 있는 게 얼마 되지 않는 것 같아도
43두 번 다신 내게 부끄럽지 않게
44똑바로 봐
45
46굳이 봐줄 필요 없어 어차피 never give up
47뭐가 덤벼오든 나는 어쩌라 하듯이
48뭐든 이길 수 있을거야
49
50이젠 깨어나 설레는 이 마음 따라
51기다리던 오늘을 만들 거야
52
53도시 사이에 피어나 이겨내온 꽃처럼
54분명 피워낼 거야 나를
55
56time’s up 몇 번을 해도 Overcharge
57그만해도 된다 해도 (그만해도 된다 해도)
58no matter no matter 숨이 차올라도
59이 길을 바란다면
60
61하늘 저 너머 뻗어가는 빛 따라
62이 마음도 날려 곧 닿을 것만 같아
63
64도시 사이에 피어나 이겨내온 꽃처럼
65분명 피워낼 거야 나를"""
66
67embed1 = model.encode(text)
68embed2 = np.array(llama.embed(text), dtype=np.float32)
69print(cos_sim(embed1, embed2).item())0.9999635815620422