Commit 5b0acfe0 authored by gd1551's avatar gd1551 Committed by GitHub

change knn config separator

parent 561ecafa
......@@ -711,7 +711,7 @@ class EmbedderModel(nn.Module):
self.model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2').cuda()
self.indexes = {}
for folder in knn_folder.split(","):
name, url = folder.split(":")
name, url = folder.split("=")
index = pickle.load(requests.get(f"{url}/index.pkl", stream='True').raw)
tag_count = pickle.load(requests.get(f"{url}/all_tags.pkl", stream='True').raw)
tag_count_sorted = sorted(tag_count.items(), key=lambda x: x[1], reverse=True)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment