POKASH.PL
/
ably.do
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mod

This commit is contained in:
l.gabrysiak 2025-02-26 15:41:53 +01:00
parent 9228f77c94
commit c1ba5ae5be
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gemma-faiss.py
View File

@ -40,7 +40,7 @@ index = faiss.IndexFlatL2(dim)
index.add(np.array(embeddings, dtype=np.float32)) index.add(np.array(embeddings, dtype=np.float32))
# 5️⃣ Wczytanie modelu Ollama (Gemma 2) # 5️⃣ Wczytanie modelu Ollama (Gemma 2)
model_name = "./trained_model/gemma/Gemma-F16-LoRA.gguf" # Ścieżka do modelu w systemie model_name = "hse.ably.do:latest" # Ścieżka do modelu w systemie
tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b") tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b")
# 6️⃣ Funkcja wyszukiwania w FAISS # 6️⃣ Funkcja wyszukiwania w FAISS