Apply isort and black reformatting

Signed-off-by: yaoyu-33 <[email protected]>
NVIDIA · Nov 26, 2024 · d9f9a4c · d9f9a4c
1 parent 9cdf6ae
commit d9f9a4c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/nemo/collections/vlm/recipes/llava15_13b.py b/nemo/collections/vlm/recipes/llava15_13b.py
@@ -18,6 +18,7 @@
 import nemo_run as run
 import pytorch_lightning as pl
 import torch
+from megatron.core.distributed import DistributedDataParallelConfig
 
 from nemo import lightning as nl
 from nemo.collections import llm, vlm
@@ -26,10 +27,8 @@
 from nemo.collections.llm.recipes.optim.adam import distributed_fused_adam_with_cosine_annealing
 from nemo.collections.llm.recipes.precision.mixed_precision import bf16_mixed
 from nemo.collections.vlm.neva.data.mock import MockDataModule
-from nemo.utils.exp_manager import TimingCallback
 from nemo.lightning.pytorch.callbacks.megatron_comm_overlap import MegatronCommOverlapCallback
-
-from megatron.core.distributed import DistributedDataParallelConfig
+from nemo.utils.exp_manager import TimingCallback
 
 NAME = "llava15_13b"
 
@@ -105,7 +104,7 @@ def finetune_recipe(
             overlap_grad_reduce=True,
             overlap_param_gather=True,
             average_in_collective=True,
-        )
+        ),
     )
 
     trainer = run.Config(

diff --git a/nemo/collections/vlm/recipes/llava15_7b.py b/nemo/collections/vlm/recipes/llava15_7b.py
@@ -18,6 +18,7 @@
 import nemo_run as run
 import pytorch_lightning as pl
 import torch
+from megatron.core.distributed import DistributedDataParallelConfig
 
 from nemo import lightning as nl
 from nemo.collections import llm, vlm
@@ -26,10 +27,8 @@
 from nemo.collections.llm.recipes.optim.adam import distributed_fused_adam_with_cosine_annealing
 from nemo.collections.llm.recipes.precision.mixed_precision import bf16_mixed
 from nemo.collections.vlm.neva.data.mock import MockDataModule
-from nemo.utils.exp_manager import TimingCallback
 from nemo.lightning.pytorch.callbacks.megatron_comm_overlap import MegatronCommOverlapCallback
-
-from megatron.core.distributed import DistributedDataParallelConfig
+from nemo.utils.exp_manager import TimingCallback
 
 NAME = "llava15_7b"
 
@@ -105,7 +104,7 @@ def finetune_recipe(
             overlap_grad_reduce=True,
             overlap_param_gather=True,
             average_in_collective=True,
-        )
+        ),
     )
 
     trainer = run.Config(