seanpedrickcase
commited on
Commit
•
1dc162b
1
Parent(s):
fea085c
embedding files now write to output folder
Browse files
search_funcs/semantic_functions.py
CHANGED
@@ -126,10 +126,10 @@ def docs_to_bge_embed_np_array(docs_out, in_file, embeddings_state, output_file_
|
|
126 |
|
127 |
progress(0.9, desc = "Saving embeddings to file")
|
128 |
if embeddings_super_compress == "No":
|
129 |
-
semantic_search_file_name = data_file_name_no_ext + '_bge_embeddings.npz'
|
130 |
np.savez_compressed(semantic_search_file_name, embeddings_out)
|
131 |
else:
|
132 |
-
semantic_search_file_name = data_file_name_no_ext + '_bge_embedding_compress.npz'
|
133 |
embeddings_out_round = np.round(embeddings_out, 3)
|
134 |
embeddings_out_round *= 100 # Rounding not currently used
|
135 |
np.savez_compressed(semantic_search_file_name, embeddings_out_round)
|
|
|
126 |
|
127 |
progress(0.9, desc = "Saving embeddings to file")
|
128 |
if embeddings_super_compress == "No":
|
129 |
+
semantic_search_file_name = output_folder + data_file_name_no_ext + '_bge_embeddings.npz'
|
130 |
np.savez_compressed(semantic_search_file_name, embeddings_out)
|
131 |
else:
|
132 |
+
semantic_search_file_name = output_folder + data_file_name_no_ext + '_bge_embedding_compress.npz'
|
133 |
embeddings_out_round = np.round(embeddings_out, 3)
|
134 |
embeddings_out_round *= 100 # Rounding not currently used
|
135 |
np.savez_compressed(semantic_search_file_name, embeddings_out_round)
|