From 75a46dd73e38d04831b21b4aa41a1f525b79633f Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Thu, 5 Sep 2024 23:08:51 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 examples/mm/stable_diffusion/anneal_sdxl.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/mm/stable_diffusion/anneal_sdxl.py b/examples/mm/stable_diffusion/anneal_sdxl.py
index 515479ace..64e032b12 100644
--- 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),
             )