From 2c18d29613d22b8b74e07b929c079c34d010a80e Mon Sep 17 00:00:00 2001 From: Maw-Fox Date: Tue, 15 Nov 2022 06:42:14 -0700 Subject: [PATCH] Fix from upstream merge. --- trainer/diffusers_trainer.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/trainer/diffusers_trainer.py b/trainer/diffusers_trainer.py index dc2508a..d8e9f47 100644 --- a/trainer/diffusers_trainer.py +++ b/trainer/diffusers_trainer.py @@ -86,6 +86,10 @@ parser.add_argument('--clip_penultimate', type=bool_t, default='False', help='Us parser.add_argument('--output_bucket_info', type=bool_t, default='False', help='Outputs bucket information and exits') parser.add_argument('--resize', type=bool_t, default='False', help="Resizes dataset's images to the appropriate bucket dimensions.") parser.add_argument('--use_xformers', type=bool_t, default='False', help='Use memory efficient attention') +parser.add_argument('--extended_validation', action='store_true', help='Perform extended validation of images to catch truncated or corrupt images.') +parser.add_argument('--no_migration', action='store_true', help='Do not perform migration of dataset while the `--resize` flag is active. Migration creates an adjacent folder to the dataset with _cropped.') +parser.add_argument('--skip_validation', action='store_true', help='Skip validation of images, useful for speeding up loading of very large datasets that have already been validated.') + args = parser.parse_args() def setup():