no message

model/retriever.py

@@ -23,9 +23,9 @@ class Retriever:
 
     def compute_embeddings(self):
         self.model = SentenceTransformer('all-MiniLM-L6-v2')
-        # tokenizer = self.model._first_module().tokenizer
-        # if tokenizer.pad_token is None:
-        #     tokenizer.pad_token = tokenizer.eos_token
+        tokenizer = self.model._first_module().tokenizer
+        if tokenizer.pad_token is None:
+            tokenizer.pad_token = tokenizer.eos_token
         self.chunk_embeddings = self.model.encode(self.corpus, convert_to_tensor=True)
 
     def chunk_text(self, text, chunk_size=CONFIG['CHUNK_SIZE']):