Fix typos in ColossalAI example (#2001)

fix typos

Fix typos in ColossalAI example (#2001)
fix typos
1e651ca2 · Haofan Wang · GitHub · 522f8aa7 · 1e651ca2
Unverified Commit 1e651ca2 authored Jan 16, 2023 by Haofan Wang Committed by GitHub Jan 16, 2023
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

examples/research_projects/colossalai/train_dreambooth_colossalai.py ...search_projects/colossalai/train_dreambooth_colossalai.py +5 -0

No files found.
--- a/examples/research_projects/colossalai/train_dreambooth_colossalai.py
+++ b/examples/research_projects/colossalai/train_dreambooth_colossalai.py
@@ -365,6 +365,11 @@ def gemini_zero_dpp(model: torch.nn.Module, placememt_policy: str = "auto"):


 def main(args):
+    if args.seed is None:
+        colossalai.launch_from_torch(config={})
+    else:
+        colossalai.launch_from_torch(config={}, seed=args.seed)
+
    colossalai.launch_from_torch(config={})

    if args.seed is not None: