Merge branch 'main' of https://github.com/NVIDIA/Megatron-LM

0024a5c6 · zhuwenwen · b004456b · 3db2063b · 0024a5c6 · 0024a5c6
Commit 0024a5c6 authored Apr 26, 2023 by zhuwenwen
12 changed files
--- a/tools/retro/index/indexes/__init__.py
+++ b/tools/retro/index/indexes/__init__.py
--- a/tools/retro/index/indexes/faiss_base.py
+++ b/tools/retro/index/indexes/faiss_base.py
--- a/tools/retro/index/indexes/faiss_par_add.py
+++ b/tools/retro/index/indexes/faiss_par_add.py
--- a/tools/retro/index/utils.py
+++ b/tools/retro/index/utils.py
--- a/tools/retro/main.py
+++ b/tools/retro/main.py
--- a/tools/retro/pretraining/chunk_dataset.py
+++ b/tools/retro/pretraining/chunk_dataset.py
--- a/tools/retro/pretraining/query.py
+++ b/tools/retro/pretraining/query.py
--- a/tools/retro/pretraining/retro_dataset.py
+++ b/tools/retro/pretraining/retro_dataset.py
--- a/tools/retro/pretraining/utils.py
+++ b/tools/retro/pretraining/utils.py
--- a/tools/retro/utils.py
+++ b/tools/retro/utils.py
--- a/tools/run_text_generation_server.py
+++ b/tools/run_text_generation_server.py
--- a/tools/text_generation_cli.py
+++ b/tools/text_generation_cli.py