From 63dd8da31a8f142eda0686516ee31436f8daadeb Mon Sep 17 00:00:00 2001 From: guyueh1 <140554423+guyueh1@users.noreply.github.com> Date: Thu, 5 Dec 2024 13:24:33 -0800 Subject: [PATCH] Handle exception when importing RetroGPTChunkDatasets (#11415) * Handle exception when importing RetroGPTChunkDatasets Signed-off-by: Guyue Huang * Update baseline Signed-off-by: guyueh1 * Revert "Update baseline" This reverts commit 21a4734c99978c0d642d44ea95f06eb8ad213a5e. * Update baseline Signed-off-by: guyueh1 * Revert "Update baseline" This reverts commit 8d35fccf72e93c8cd6807d34007d991703e077d8. --------- Signed-off-by: Guyue Huang Signed-off-by: guyueh1 Co-authored-by: guyueh1 --- .../nlp/data/language_modeling/megatron/retro_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nemo/collections/nlp/data/language_modeling/megatron/retro_dataset.py b/nemo/collections/nlp/data/language_modeling/megatron/retro_dataset.py index d8dafd69c658..e064dbf14d5d 100644 --- a/nemo/collections/nlp/data/language_modeling/megatron/retro_dataset.py +++ b/nemo/collections/nlp/data/language_modeling/megatron/retro_dataset.py @@ -47,7 +47,7 @@ HAVE_TE_AND_MEGATRON_CORE = True -except (ImportError, ModuleNotFoundError): +except (ImportError, ModuleNotFoundError, Exception): HAVE_TE_AND_MEGATRON_CORE = False from typing import Any