From c0a78361361f8fe03dda101bc3def875bb80a775 Mon Sep 17 00:00:00 2001 From: suiyoubi Date: Mon, 2 Dec 2024 16:01:51 +0000 Subject: [PATCH] Apply isort and black reformatting Signed-off-by: suiyoubi --- nemo/collections/llm/bert/model/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nemo/collections/llm/bert/model/base.py b/nemo/collections/llm/bert/model/base.py index c55ae1108845..8c11dfc0500c 100644 --- a/nemo/collections/llm/bert/model/base.py +++ b/nemo/collections/llm/bert/model/base.py @@ -48,7 +48,7 @@ from megatron.core.models.bert.bert_model import BertModel as MCoreBert except (ImportError, ModuleNotFoundError) as e: HAVE_TE = False - MCoreBert = TransformerLayer # Place holder for import checking. BERT requires TE installed. + MCoreBert = TransformerLayer # Place holder for import checking. BERT requires TE installed. if TYPE_CHECKING: from nemo.collections.common.tokenizers.tokenizer_spec import TokenizerSpec