seanpedrickcase commited on
Commit
1dc162b
1 Parent(s): fea085c

embedding files now write to output folder

Browse files
Files changed (1) hide show
  1. search_funcs/semantic_functions.py +2 -2
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)