Skip to content

Commit 9c48ce3

Browse files
authored
Add files via upload
1 parent 4eb4412 commit 9c48ce3

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/database_interactions.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -302,6 +302,7 @@ def initialize_vector_model(self, embedding_model_name, config_data):
302302
'batch_size': 8,
303303
'padding': True,
304304
'truncation': True,
305+
"normalize_embeddings": True,
305306
}
306307

307308
if compute_device.lower() == 'cpu':
@@ -670,6 +671,8 @@ def initialize_vector_model(self):
670671

671672
encode_kwargs = {
672673
"batch_size": 1,
674+
'padding': True,
675+
'truncation': True,
673676
"normalize_embeddings": True,
674677
}
675678

0 commit comments

Comments
 (0)