From 8e1d4f341431cb82291fef846415e28601c3026b Mon Sep 17 00:00:00 2001 From: pomchi Date: Wed, 13 Dec 2023 10:28:52 +0800 Subject: [PATCH] merge pull request #1000 --- library/train_util.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/library/train_util.py b/library/train_util.py index 71cf49f35..625ba8618 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -1722,7 +1722,6 @@ def __init__( subset.caption_extension, subset.num_repeats, subset.shuffle_caption, - subset.caption_separator, subset.keep_tokens, subset.color_aug, subset.flip_aug, @@ -3838,7 +3837,7 @@ def prepare_accelerator(args: argparse.Namespace): if args.wandb_api_key is not None: wandb.login(key=args.wandb_api_key) - + kwargs_handlers = ( InitProcessGroupKwargs(timeout=datetime.timedelta(minutes=args.ddp_timeout)) if args.ddp_timeout else None, DistributedDataParallelKwargs(gradient_as_bucket_view=args.gradient_as_bucket_view, static_graph=args.static_graph)