Apply isort and black reformatting

Signed-off-by: akoumpa <[email protected]>

nemo/collections/llm/gpt/model/auto_model.py

@@ -20,22 +20,22 @@
 import torch.nn.functional as F
 from torch import nn
 from typing_extensions import Annotated
-from nemo.lightning.pytorch.optim import OptimizerModule
-from pathlib import Path
+
+from nemo.collections.llm.gpt.model.baichuan import Baichuan2Model
 
 # import nemo.collections.llm.gpt.model as llm
 from nemo.collections.llm.gpt.model.base import GPTModel
 from nemo.collections.llm.gpt.model.chatglm import ChatGLMModel
-from nemo.collections.llm.gpt.model.phi3mini import Phi3Model
-from nemo.collections.llm.gpt.model.mixtral import MixtralModel
-from nemo.collections.llm.gpt.model.mistral import MistralModel
 from nemo.collections.llm.gpt.model.gemma import GemmaModel
 from nemo.collections.llm.gpt.model.gemma2 import Gemma2Model
 from nemo.collections.llm.gpt.model.llama import LlamaModel
-from nemo.collections.llm.gpt.model.baichuan import Baichuan2Model
+from nemo.collections.llm.gpt.model.mistral import MistralModel
+from nemo.collections.llm.gpt.model.mixtral import MixtralModel
+from nemo.collections.llm.gpt.model.phi3mini import Phi3Model
 from nemo.collections.llm.gpt.model.qwen2 import Qwen2Model
 from nemo.collections.llm.gpt.model.starcoder import StarcoderModel
 from nemo.collections.llm.gpt.model.starcoder2 import Starcoder2Model
+from nemo.lightning.pytorch.optim import OptimizerModule
 
 HF_TO_MCORE_REGISTRY = {
     'ChatGLMForCausalLM': ChatGLMModel,
@@ -73,6 +73,7 @@ def __init__(
 
         # Get model class from registry
         from transformers import AutoConfig
+
         architectures = AutoConfig.from_pretrained(model_name_or_path, trust_remote_code=True).architectures
         assert isinstance(architectures, list), "Expected architectures to be a list"
         assert len(architectures) == 1, "Expected architectures to contain one item"
@@ -90,13 +91,12 @@ def __init__(
         config = model_cls.importer(import_path).config
 
         # Init class
-        super().__init__(
-            config, optim=optim, tokenizer=tokenizer, model_transform=model_transform
-        )
+        super().__init__(config, optim=optim, tokenizer=tokenizer, model_transform=model_transform)
 
         # Change self's class to model_cls
         self.__class__ = model_cls
 
+
 __all__ = [
     "AutoModel",
 ]