From b895fda3b0686b47adb8e77de38822f2b0061ebc Mon Sep 17 00:00:00 2001
From: "l.gabrysiak" <l.gabrysiak@osadkowski.pl>
Date: Fri, 28 Feb 2025 21:28:33 +0100
Subject: [PATCH] mod allegro

---
 allegro.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/allegro.py b/allegro.py
index 54bc777..b7d0b99 100644
--- a/allegro.py
+++ b/allegro.py
@@ -1,11 +1,12 @@
 import os
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
-import torch
-import faiss
 from weaviate.connect import ConnectionParams
 import weaviate
 
+import torch
+import faiss
+import numpy as np
 from sentence_transformers import SentenceTransformer
 from datasets import Dataset
 from peft import LoraConfig, get_peft_model