From 73cd072c42641c3e9be803a337be1d9331fffdeb Mon Sep 17 00:00:00 2001 From: William Berman Date: Thu, 16 Feb 2023 00:01:21 -0800 Subject: [PATCH] checkpointing_steps_total_limit->checkpoints_total_limit --- examples/dreambooth/train_dreambooth.py | 4 ++-- examples/dreambooth/train_dreambooth_lora.py | 4 ++-- .../dreambooth_inpaint/train_dreambooth_inpaint.py | 4 ++-- .../dreambooth_inpaint/train_dreambooth_inpaint_lora.py | 4 ++-- .../train_multi_subject_dreambooth.py | 4 ++-- .../onnxruntime/text_to_image/train_text_to_image.py | 4 ++-- .../onnxruntime/textual_inversion/textual_inversion.py | 4 ++-- .../unconditional_image_generation/train_unconditional.py | 4 ++-- examples/text_to_image/train_text_to_image.py | 4 ++-- examples/text_to_image/train_text_to_image_lora.py | 4 ++-- examples/textual_inversion/textual_inversion.py | 4 ++-- .../unconditional_image_generation/train_unconditional.py | 4 ++-- 12 files changed, 24 insertions(+), 24 deletions(-) diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py index 955f6ec9547e..4fd0e5a08ce0 100644 --- a/examples/dreambooth/train_dreambooth.py +++ b/examples/dreambooth/train_dreambooth.py @@ -196,7 +196,7 @@ def parse_args(input_args=None): ), ) parser.add_argument( - "--checkpointing_steps_total_limit", + "--checkpoints_total_limit", type=int, default=None, help=( @@ -498,7 +498,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token: def main(args): logging_dir = Path(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) + accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, diff --git a/examples/dreambooth/train_dreambooth_lora.py b/examples/dreambooth/train_dreambooth_lora.py index 52951577ad34..f262b71eb006 100644 --- a/examples/dreambooth/train_dreambooth_lora.py +++ b/examples/dreambooth/train_dreambooth_lora.py @@ -243,7 +243,7 @@ def parse_args(input_args=None): ), ) parser.add_argument( - "--checkpointing_steps_total_limit", + "--checkpoints_total_limit", type=int, default=None, help=( @@ -536,7 +536,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token: def main(args): logging_dir = Path(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) + accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, diff --git a/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py b/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py index c8dc819b2631..247361d21299 100644 --- a/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py +++ b/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py @@ -259,7 +259,7 @@ def parse_args(): ), ) parser.add_argument( - "--checkpointing_steps_total_limit", + "--checkpoints_total_limit", type=int, default=None, help=( @@ -416,7 +416,7 @@ def main(): args = parse_args() logging_dir = Path(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) + accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, diff --git a/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py b/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py index 2c6f1bc46581..4acc6e501b32 100644 --- a/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py +++ b/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py @@ -255,7 +255,7 @@ def parse_args(): ), ) parser.add_argument( - "--checkpointing_steps_total_limit", + "--checkpoints_total_limit", type=int, default=None, help=( @@ -415,7 +415,7 @@ def main(): args = parse_args() logging_dir = Path(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) + accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, diff --git a/examples/research_projects/multi_subject_dreambooth/train_multi_subject_dreambooth.py b/examples/research_projects/multi_subject_dreambooth/train_multi_subject_dreambooth.py index 85bacc7aa2e4..2ea6217e576f 100644 --- a/examples/research_projects/multi_subject_dreambooth/train_multi_subject_dreambooth.py +++ b/examples/research_projects/multi_subject_dreambooth/train_multi_subject_dreambooth.py @@ -171,7 +171,7 @@ def parse_args(input_args=None): ), ) parser.add_argument( - "--checkpointing_steps_total_limit", + "--checkpoints_total_limit", type=int, default=None, help=( @@ -476,7 +476,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token: def main(args): logging_dir = Path(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) + accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, diff --git a/examples/research_projects/onnxruntime/text_to_image/train_text_to_image.py b/examples/research_projects/onnxruntime/text_to_image/train_text_to_image.py index c21e152004fa..2ec7dcf1548c 100644 --- a/examples/research_projects/onnxruntime/text_to_image/train_text_to_image.py +++ b/examples/research_projects/onnxruntime/text_to_image/train_text_to_image.py @@ -275,7 +275,7 @@ def parse_args(): ), ) parser.add_argument( - "--checkpointing_steps_total_limit", + "--checkpoints_total_limit", type=int, default=None, help=( @@ -332,7 +332,7 @@ def main(): args = parse_args() logging_dir = os.path.join(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) + accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, diff --git a/examples/research_projects/onnxruntime/textual_inversion/textual_inversion.py b/examples/research_projects/onnxruntime/textual_inversion/textual_inversion.py index 100d742b9aca..d6faae19416a 100644 --- a/examples/research_projects/onnxruntime/textual_inversion/textual_inversion.py +++ b/examples/research_projects/onnxruntime/textual_inversion/textual_inversion.py @@ -291,7 +291,7 @@ def parse_args(): ), ) parser.add_argument( - "--checkpointing_steps_total_limit", + "--checkpoints_total_limit", type=int, default=None, help=( @@ -477,7 +477,7 @@ def main(): args = parse_args() logging_dir = os.path.join(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) + accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, diff --git a/examples/research_projects/onnxruntime/unconditional_image_generation/train_unconditional.py b/examples/research_projects/onnxruntime/unconditional_image_generation/train_unconditional.py index 429f3eac4faf..1b38036d82c0 100644 --- a/examples/research_projects/onnxruntime/unconditional_image_generation/train_unconditional.py +++ b/examples/research_projects/onnxruntime/unconditional_image_generation/train_unconditional.py @@ -233,7 +233,7 @@ def parse_args(): ), ) parser.add_argument( - "--checkpointing_steps_total_limit", + "--checkpoints_total_limit", type=int, default=None, help=( @@ -276,7 +276,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token: def main(args): logging_dir = os.path.join(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) + accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, diff --git a/examples/text_to_image/train_text_to_image.py b/examples/text_to_image/train_text_to_image.py index f915bdb9cfae..5a6d83d5bb41 100644 --- a/examples/text_to_image/train_text_to_image.py +++ b/examples/text_to_image/train_text_to_image.py @@ -276,7 +276,7 @@ def parse_args(): ), ) parser.add_argument( - "--checkpointing_steps_total_limit", + "--checkpoints_total_limit", type=int, default=None, help=( @@ -343,7 +343,7 @@ def main(): ) logging_dir = os.path.join(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) + accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, diff --git a/examples/text_to_image/train_text_to_image_lora.py b/examples/text_to_image/train_text_to_image_lora.py index 69c978aca518..0b82a5a78f43 100644 --- a/examples/text_to_image/train_text_to_image_lora.py +++ b/examples/text_to_image/train_text_to_image_lora.py @@ -311,7 +311,7 @@ def parse_args(): ), ) parser.add_argument( - "--checkpointing_steps_total_limit", + "--checkpoints_total_limit", type=int, default=None, help=( @@ -364,7 +364,7 @@ def main(): args = parse_args() logging_dir = os.path.join(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) + accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, diff --git a/examples/textual_inversion/textual_inversion.py b/examples/textual_inversion/textual_inversion.py index 9117ca5336db..22c570fb9e6d 100644 --- a/examples/textual_inversion/textual_inversion.py +++ b/examples/textual_inversion/textual_inversion.py @@ -289,7 +289,7 @@ def parse_args(): ), ) parser.add_argument( - "--checkpointing_steps_total_limit", + "--checkpoints_total_limit", type=int, default=None, help=( @@ -475,7 +475,7 @@ def main(): args = parse_args() logging_dir = os.path.join(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) + accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, diff --git a/examples/unconditional_image_generation/train_unconditional.py b/examples/unconditional_image_generation/train_unconditional.py index 4b8c8fccdbc7..98af8dc354c5 100644 --- a/examples/unconditional_image_generation/train_unconditional.py +++ b/examples/unconditional_image_generation/train_unconditional.py @@ -241,7 +241,7 @@ def parse_args(): ), ) parser.add_argument( - "--checkpointing_steps_total_limit", + "--checkpoints_total_limit", type=int, default=None, help=( @@ -284,7 +284,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token: def main(args): logging_dir = os.path.join(args.output_dir, args.logging_dir) - accelerator_project_config = ProjectConfiguration(total_limit=args.checkpointing_steps_total_limit) + accelerator_project_config = ProjectConfiguration(total_limit=args.checkpoints_total_limit) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps,