Apply isort and black reformatting

Signed-off-by: akoumpa <akoumpa@users.noreply.github.com>
NVIDIA · Sep 24, 2024 · 03c1176 · 03c1176
1 parent 0ae8380
commit 03c1176
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/nemo/collections/llm/__init__.py b/nemo/collections/llm/__init__.py
@@ -18,7 +18,6 @@
 safe_import("transformer_engine")
 
 from nemo.collections.llm import peft, tokenizer
-
 from nemo.collections.llm.gpt.data import (
     DollyDataModule,
     FineTuningDataModule,
@@ -70,8 +69,8 @@
     LlamaModel,
     MaskedTokenLossReduction,
     MistralConfig7B,
-    MistralNeMo2407Config12B,
     MistralModel,
+    MistralNeMo2407Config12B,
     MixtralConfig8x3B,
     MixtralConfig8x7B,
     MixtralConfig8x22B,
@@ -104,7 +103,6 @@
     gpt_forward_step,
 )
 
-
 __all__ = [
     "MockDataModule",
     "GPTModel",
@@ -186,6 +184,7 @@
 
 try:
     import nemo_run as run
+
     from nemo.collections.llm.api import export_ckpt, finetune, import_ckpt, pretrain, train, validate
     from nemo.collections.llm.recipes import *  # noqa
 

diff --git a/nemo/collections/llm/gpt/model/__init__.py b/nemo/collections/llm/gpt/model/__init__.py
@@ -53,7 +53,7 @@
     LlamaConfig,
     LlamaModel,
 )
-from nemo.collections.llm.gpt.model.mistral import MistralConfig7B, MistralNeMo2407Config12B, MistralModel
+from nemo.collections.llm.gpt.model.mistral import MistralConfig7B, MistralModel, MistralNeMo2407Config12B
 from nemo.collections.llm.gpt.model.mixtral import (
     MixtralConfig8x3B,
     MixtralConfig8x7B,