From 5787f7d080aa5e5cea9bb3d664a57851137e1f44 Mon Sep 17 00:00:00 2001 From: laksjdjf Date: Mon, 21 Nov 2022 13:57:26 +0900 Subject: [PATCH] Update diffusers_trainer.py --- trainer/diffusers_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trainer/diffusers_trainer.py b/trainer/diffusers_trainer.py index 30cedb1..551171c 100644 --- a/trainer/diffusers_trainer.py +++ b/trainer/diffusers_trainer.py @@ -719,7 +719,7 @@ def main(): # load dataset store = ImageStore(args.dataset) - dataset = AspectDataset(store, tokenizer) + dataset = AspectDataset(store, tokenizer, ucg=args.ucg) bucket = AspectBucket(store, args.num_buckets, args.batch_size, args.bucket_side_min, args.bucket_side_max, 64, args.resolution * args.resolution, 2.0) sampler = AspectBucketSampler(bucket=bucket, num_replicas=world_size, rank=rank)