summaryrefslogtreecommitdiffstats
path: root/train_dreambooth.py
diff options
context:
space:
mode:
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r--train_dreambooth.py4
1 files changed, 0 insertions, 4 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py
index cd0bf67..05f6cb5 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -955,10 +955,6 @@ def main():
955 ) 955 )
956 global_progress_bar.set_description("Total progress") 956 global_progress_bar.set_description("Total progress")
957 957
958 save_args(basepath, args, {
959 "global_step": global_step + global_step_offset
960 })
961
962 try: 958 try:
963 for epoch in range(num_epochs): 959 for epoch in range(num_epochs):
964 if accelerator.is_main_process: 960 if accelerator.is_main_process: