Skip to content

GitLab

Unverified Commit 1e651ca2 authored by Haofan Wang's avatar Haofan Wang Committed by GitHub
Browse files

Fix typos in ColossalAI example (#2001) 

fix typos
parent 522f8aa7
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
examples/research_projects/colossalai/train_dreambooth_colossalai.py
View file @ 1e651ca2
...@@ -365,6 +365,11 @@ def gemini_zero_dpp(model: torch.nn.Module, placememt_policy: str = "auto"): ...@@ -365,6 +365,11 @@ def gemini_zero_dpp(model: torch.nn.Module, placememt_policy: str = "auto"):
def main(args): 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={}) colossalai.launch_from_torch(config={})
if args.seed is not None: if args.seed is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment