From 30fea8fdfaae6cc6e385d65634956d1c99db0336 Mon Sep 17 00:00:00 2001 From: SY573M_404 <40136012+SY573M404@users.noreply.github.com> Date: Sat, 11 Feb 2023 17:28:14 +0500 Subject: [PATCH] Fix typo in diffusers_trainer.py One description was used for two CLI flags --- 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 0e08746..33fc0df 100644 --- a/trainer/diffusers_trainer.py +++ b/trainer/diffusers_trainer.py @@ -81,7 +81,7 @@ parser.add_argument('--fp16', dest='fp16', type=bool_t, default='False', help='T parser.add_argument('--image_log_steps', type=int, default=500, help='Number of steps to log images at.') parser.add_argument('--image_log_amount', type=int, default=4, help='Number of images to log every image_log_steps') parser.add_argument('--image_log_inference_steps', type=int, default=50, help='Number of inference steps to use to log images.') -parser.add_argument('--image_log_scheduler', type=str, default="PNDMScheduler", help='Number of inference steps to use to log images.') +parser.add_argument('--image_log_scheduler', type=str, default="PNDMScheduler", help='Scheduler to use to log images.') parser.add_argument('--clip_penultimate', type=bool_t, default='False', help='Use penultimate CLIP layer for text embedding') 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.")