Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
46def726
"references/vscode:/vscode.git/clone" did not exist on "5bb81c8e008c601237e2ffd5cbea7192775130bd"
Unverified
Commit
46def726
authored
Feb 16, 2023
by
Will Berman
Committed by
GitHub
Feb 16, 2023
Browse files
checkpointing_steps_total_limit->checkpoints_total_limit (#2374)
parent
296b01e1
Changes
12
Show whitespace changes
Inline
Side-by-side
Showing
12 changed files
with
24 additions
and
24 deletions
+24
-24
examples/dreambooth/train_dreambooth.py
examples/dreambooth/train_dreambooth.py
+2
-2
examples/dreambooth/train_dreambooth_lora.py
examples/dreambooth/train_dreambooth_lora.py
+2
-2
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py
...h_projects/dreambooth_inpaint/train_dreambooth_inpaint.py
+2
-2
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py
...jects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py
+2
-2
examples/research_projects/multi_subject_dreambooth/train_multi_subject_dreambooth.py
...ulti_subject_dreambooth/train_multi_subject_dreambooth.py
+2
-2
examples/research_projects/onnxruntime/text_to_image/train_text_to_image.py
...projects/onnxruntime/text_to_image/train_text_to_image.py
+2
-2
examples/research_projects/onnxruntime/textual_inversion/textual_inversion.py
...ojects/onnxruntime/textual_inversion/textual_inversion.py
+2
-2
examples/research_projects/onnxruntime/unconditional_image_generation/train_unconditional.py
...ime/unconditional_image_generation/train_unconditional.py
+2
-2
examples/text_to_image/train_text_to_image.py
examples/text_to_image/train_text_to_image.py
+2
-2
examples/text_to_image/train_text_to_image_lora.py
examples/text_to_image/train_text_to_image_lora.py
+2
-2
examples/textual_inversion/textual_inversion.py
examples/textual_inversion/textual_inversion.py
+2
-2
examples/unconditional_image_generation/train_unconditional.py
...les/unconditional_image_generation/train_unconditional.py
+2
-2
No files found.
examples/dreambooth/train_dreambooth.py
View file @
46def726
...
@@ -196,7 +196,7 @@ def parse_args(input_args=None):
...
@@ -196,7 +196,7 @@ def parse_args(input_args=None):
),
),
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--checkpoint
ing_step
s_total_limit"
,
"--checkpoints_total_limit"
,
type
=
int
,
type
=
int
,
default
=
None
,
default
=
None
,
help
=
(
help
=
(
...
@@ -498,7 +498,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
...
@@ -498,7 +498,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def
main
(
args
):
def
main
(
args
):
logging_dir
=
Path
(
args
.
output_dir
,
args
.
logging_dir
)
logging_dir
=
Path
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoint
ing_step
s_total_limit
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
...
...
examples/dreambooth/train_dreambooth_lora.py
View file @
46def726
...
@@ -243,7 +243,7 @@ def parse_args(input_args=None):
...
@@ -243,7 +243,7 @@ def parse_args(input_args=None):
),
),
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--checkpoint
ing_step
s_total_limit"
,
"--checkpoints_total_limit"
,
type
=
int
,
type
=
int
,
default
=
None
,
default
=
None
,
help
=
(
help
=
(
...
@@ -536,7 +536,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
...
@@ -536,7 +536,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def
main
(
args
):
def
main
(
args
):
logging_dir
=
Path
(
args
.
output_dir
,
args
.
logging_dir
)
logging_dir
=
Path
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoint
ing_step
s_total_limit
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
...
...
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py
View file @
46def726
...
@@ -259,7 +259,7 @@ def parse_args():
...
@@ -259,7 +259,7 @@ def parse_args():
),
),
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--checkpoint
ing_step
s_total_limit"
,
"--checkpoints_total_limit"
,
type
=
int
,
type
=
int
,
default
=
None
,
default
=
None
,
help
=
(
help
=
(
...
@@ -416,7 +416,7 @@ def main():
...
@@ -416,7 +416,7 @@ def main():
args
=
parse_args
()
args
=
parse_args
()
logging_dir
=
Path
(
args
.
output_dir
,
args
.
logging_dir
)
logging_dir
=
Path
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoint
ing_step
s_total_limit
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
...
...
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py
View file @
46def726
...
@@ -255,7 +255,7 @@ def parse_args():
...
@@ -255,7 +255,7 @@ def parse_args():
),
),
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--checkpoint
ing_step
s_total_limit"
,
"--checkpoints_total_limit"
,
type
=
int
,
type
=
int
,
default
=
None
,
default
=
None
,
help
=
(
help
=
(
...
@@ -415,7 +415,7 @@ def main():
...
@@ -415,7 +415,7 @@ def main():
args
=
parse_args
()
args
=
parse_args
()
logging_dir
=
Path
(
args
.
output_dir
,
args
.
logging_dir
)
logging_dir
=
Path
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoint
ing_step
s_total_limit
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
...
...
examples/research_projects/multi_subject_dreambooth/train_multi_subject_dreambooth.py
View file @
46def726
...
@@ -171,7 +171,7 @@ def parse_args(input_args=None):
...
@@ -171,7 +171,7 @@ def parse_args(input_args=None):
),
),
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--checkpoint
ing_step
s_total_limit"
,
"--checkpoints_total_limit"
,
type
=
int
,
type
=
int
,
default
=
None
,
default
=
None
,
help
=
(
help
=
(
...
@@ -476,7 +476,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
...
@@ -476,7 +476,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def
main
(
args
):
def
main
(
args
):
logging_dir
=
Path
(
args
.
output_dir
,
args
.
logging_dir
)
logging_dir
=
Path
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoint
ing_step
s_total_limit
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
...
...
examples/research_projects/onnxruntime/text_to_image/train_text_to_image.py
View file @
46def726
...
@@ -275,7 +275,7 @@ def parse_args():
...
@@ -275,7 +275,7 @@ def parse_args():
),
),
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--checkpoint
ing_step
s_total_limit"
,
"--checkpoints_total_limit"
,
type
=
int
,
type
=
int
,
default
=
None
,
default
=
None
,
help
=
(
help
=
(
...
@@ -332,7 +332,7 @@ def main():
...
@@ -332,7 +332,7 @@ def main():
args
=
parse_args
()
args
=
parse_args
()
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoint
ing_step
s_total_limit
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
...
...
examples/research_projects/onnxruntime/textual_inversion/textual_inversion.py
View file @
46def726
...
@@ -291,7 +291,7 @@ def parse_args():
...
@@ -291,7 +291,7 @@ def parse_args():
),
),
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--checkpoint
ing_step
s_total_limit"
,
"--checkpoints_total_limit"
,
type
=
int
,
type
=
int
,
default
=
None
,
default
=
None
,
help
=
(
help
=
(
...
@@ -477,7 +477,7 @@ def main():
...
@@ -477,7 +477,7 @@ def main():
args
=
parse_args
()
args
=
parse_args
()
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoint
ing_step
s_total_limit
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
...
...
examples/research_projects/onnxruntime/unconditional_image_generation/train_unconditional.py
View file @
46def726
...
@@ -233,7 +233,7 @@ def parse_args():
...
@@ -233,7 +233,7 @@ def parse_args():
),
),
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--checkpoint
ing_step
s_total_limit"
,
"--checkpoints_total_limit"
,
type
=
int
,
type
=
int
,
default
=
None
,
default
=
None
,
help
=
(
help
=
(
...
@@ -276,7 +276,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
...
@@ -276,7 +276,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def
main
(
args
):
def
main
(
args
):
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoint
ing_step
s_total_limit
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
...
...
examples/text_to_image/train_text_to_image.py
View file @
46def726
...
@@ -276,7 +276,7 @@ def parse_args():
...
@@ -276,7 +276,7 @@ def parse_args():
),
),
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--checkpoint
ing_step
s_total_limit"
,
"--checkpoints_total_limit"
,
type
=
int
,
type
=
int
,
default
=
None
,
default
=
None
,
help
=
(
help
=
(
...
@@ -343,7 +343,7 @@ def main():
...
@@ -343,7 +343,7 @@ def main():
)
)
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoint
ing_step
s_total_limit
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
...
...
examples/text_to_image/train_text_to_image_lora.py
View file @
46def726
...
@@ -311,7 +311,7 @@ def parse_args():
...
@@ -311,7 +311,7 @@ def parse_args():
),
),
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--checkpoint
ing_step
s_total_limit"
,
"--checkpoints_total_limit"
,
type
=
int
,
type
=
int
,
default
=
None
,
default
=
None
,
help
=
(
help
=
(
...
@@ -364,7 +364,7 @@ def main():
...
@@ -364,7 +364,7 @@ def main():
args
=
parse_args
()
args
=
parse_args
()
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoint
ing_step
s_total_limit
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
...
...
examples/textual_inversion/textual_inversion.py
View file @
46def726
...
@@ -289,7 +289,7 @@ def parse_args():
...
@@ -289,7 +289,7 @@ def parse_args():
),
),
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--checkpoint
ing_step
s_total_limit"
,
"--checkpoints_total_limit"
,
type
=
int
,
type
=
int
,
default
=
None
,
default
=
None
,
help
=
(
help
=
(
...
@@ -475,7 +475,7 @@ def main():
...
@@ -475,7 +475,7 @@ def main():
args
=
parse_args
()
args
=
parse_args
()
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoint
ing_step
s_total_limit
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
...
...
examples/unconditional_image_generation/train_unconditional.py
View file @
46def726
...
@@ -241,7 +241,7 @@ def parse_args():
...
@@ -241,7 +241,7 @@ def parse_args():
),
),
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--checkpoint
ing_step
s_total_limit"
,
"--checkpoints_total_limit"
,
type
=
int
,
type
=
int
,
default
=
None
,
default
=
None
,
help
=
(
help
=
(
...
@@ -284,7 +284,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
...
@@ -284,7 +284,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token:
def
main
(
args
):
def
main
(
args
):
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
logging_dir
=
os
.
path
.
join
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoint
ing_step
s_total_limit
)
accelerator_project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment