Skip to content

Commit 4413807

Browse files
committed
Deleting pre-computed embeddings if re-computing them
1 parent 0f50daa commit 4413807

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/thunder/benchmark.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -294,8 +294,9 @@ def run_benchmark(cfg: DictConfig, model_cls: Callable = None) -> None:
294294
)
295295

296296
if cfg.embedding_recomputing.recompute_embeddings:
297-
emb_info_str += " Re-computing them as explictly requested."
297+
emb_info_str += " Deleting pre-computed embeddings and re-computing them as explictly requested."
298298
comp_embs = True
299+
shutil.rmtree(embeddings_folder)
299300
else:
300301
emb_info_str += " Not re-computing them."
301302
comp_embs = False

0 commit comments

Comments
 (0)