Fix typos in ColossalAI example (#2001)

fix typos
2023-01-16 22:21:04 +08:00 · 2023-01-16 22:21:04 +08:00 · 1e651ca2c9
parent 522f8aa7b2
commit 1e651ca2c9
1 changed files with 5 additions and 0 deletions
--- 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: