AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Fix typos in ColossalAI example (#2001)

Browse Source 
fix typos
This commit is contained in:
Haofan Wang
2023-01-16 22:21:04 +08:00
committed by GitHub
parent 522f8aa7b2
commit 1e651ca2c9
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/research_projects/colossalai/train_dreambooth_colossalai.py
View File

@@ -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: