Merge branch 'draftp_anneal_docs' of github.com:rohitrango/NeMo-Align…

…er into draftp_anneal_docs
NVIDIA · Sep 6, 2024 · 6470842 · 6470842
2 parents de8862b + 75a46dd
commit 6470842
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/mm/stable_diffusion/anneal_sdxl.py b/examples/mm/stable_diffusion/anneal_sdxl.py
@@ -93,6 +93,7 @@
 
 mp.set_start_method("spawn", force=True)
 
+
 class MegatronStableDiffusionTrainerBuilder(MegatronTrainerBuilder):
     """Builder for SD model Trainer with overrides."""
 
@@ -130,7 +131,7 @@ def _training_strategy(self) -> NLPDDPStrategy:
                     FrozenCLIPEmbedder,
                     ParallelLinearAdapter,
                 },
-                use_orig_params=False, 
+                use_orig_params=False,
                 set_buffer_dtype=self.cfg.get("fsdp_set_buffer_dtype", None),
             )