Skip to content

Commit 1e651ca

Browse files
authored
Fix typos in ColossalAI example (huggingface#2001)
fix typos
1 parent 522f8aa commit 1e651ca

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

examples/research_projects/colossalai/train_dreambooth_colossalai.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -365,6 +365,11 @@ def gemini_zero_dpp(model: torch.nn.Module, placememt_policy: str = "auto"):
365365

366366

367367
def main(args):
368+
if args.seed is None:
369+
colossalai.launch_from_torch(config={})
370+
else:
371+
colossalai.launch_from_torch(config={}, seed=args.seed)
372+
368373
colossalai.launch_from_torch(config={})
369374

370375
if args.seed is not None:

0 commit comments

Comments
 (0)