Fix typos in ColossalAI example (huggingface#2001)

fix typos
AmericanPresidentJimmyCarter · Jan 16, 2023 · 1e651ca · 1e651ca
1 parent 522f8aa
commit 1e651ca
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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: