diff --git a/sdxl_train.py b/sdxl_train.py index 6a476a1f..724bf558 100644 --- a/sdxl_train.py +++ b/sdxl_train.py @@ -593,9 +593,7 @@ def train(args): ) # 指定ステップごとにモデルを保存 - if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and \ - args.save_every_n_steps_after_x is not None and global_step >= args.save_every_n_steps_after_x: - + if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and global_step >= args.save_every_n_steps_after_x: accelerator.wait_for_everyone() if accelerator.is_main_process: src_path = src_stable_diffusion_ckpt if save_stable_diffusion_format else src_diffusers_model_path diff --git a/sdxl_train_control_net_lllite.py b/sdxl_train_control_net_lllite.py index 9cf41c81..99abb76c 100644 --- a/sdxl_train_control_net_lllite.py +++ b/sdxl_train_control_net_lllite.py @@ -485,9 +485,7 @@ def train(args): # sdxl_train_util.sample_images(accelerator, args, None, global_step, accelerator.device, vae, tokenizer, text_encoder, unet) # 指定ステップごとにモデルを保存 - if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and \ - args.save_every_n_steps_after_x is not None and global_step >= args.save_every_n_steps_after_x: - + if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and global_step >= args.save_every_n_steps_after_x: accelerator.wait_for_everyone() if accelerator.is_main_process: ckpt_name = train_util.get_step_ckpt_name(args, "." + args.save_model_as, global_step) diff --git a/sdxl_train_control_net_lllite_old.py b/sdxl_train_control_net_lllite_old.py index 98a9818c..fde6025c 100644 --- a/sdxl_train_control_net_lllite_old.py +++ b/sdxl_train_control_net_lllite_old.py @@ -455,9 +455,7 @@ def train(args): # sdxl_train_util.sample_images(accelerator, args, None, global_step, accelerator.device, vae, tokenizer, text_encoder, unet) # 指定ステップごとにモデルを保存 - if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and \ - args.save_every_n_steps_after_x is not None and global_step >= args.save_every_n_steps_after_x: - + if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and global_step >= args.save_every_n_steps_after_x: accelerator.wait_for_everyone() if accelerator.is_main_process: ckpt_name = train_util.get_step_ckpt_name(args, "." + args.save_model_as, global_step) diff --git a/train_controlnet.py b/train_controlnet.py index 83aae7cf..6444b654 100644 --- a/train_controlnet.py +++ b/train_controlnet.py @@ -482,9 +482,7 @@ def train(args): ) # 指定ステップごとにモデルを保存 - if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and \ - args.save_every_n_steps_after_x is not None and global_step >= args.save_every_n_steps_after_x: - + if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and global_step >= args.save_every_n_steps_after_x: accelerator.wait_for_everyone() if accelerator.is_main_process: ckpt_name = train_util.get_step_ckpt_name(args, "." + args.save_model_as, global_step) diff --git a/train_db.py b/train_db.py index 831bc445..540a2dfd 100644 --- a/train_db.py +++ b/train_db.py @@ -361,9 +361,7 @@ def train(args): ) # 指定ステップごとにモデルを保存 - if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and \ - args.save_every_n_steps_after_x is not None and global_step >= args.save_every_n_steps_after_x: - + if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and global_step >= args.save_every_n_steps_after_x: accelerator.wait_for_everyone() if accelerator.is_main_process: src_path = src_stable_diffusion_ckpt if save_stable_diffusion_format else src_diffusers_model_path diff --git a/train_network.py b/train_network.py index 0f566353..7c5c5e7e 100644 --- a/train_network.py +++ b/train_network.py @@ -828,8 +828,7 @@ class NetworkTrainer: self.sample_images(accelerator, args, None, global_step, accelerator.device, vae, tokenizer, text_encoder, unet) # 指定ステップごとにモデルを保存 - if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and \ - args.save_every_n_steps_after_x is not None and global_step >= args.save_every_n_steps_after_x: + if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and global_step >= args.save_every_n_steps_after_x: accelerator.wait_for_everyone() if accelerator.is_main_process: diff --git a/train_textual_inversion.py b/train_textual_inversion.py index 9d79fad2..c09d0612 100644 --- a/train_textual_inversion.py +++ b/train_textual_inversion.py @@ -622,9 +622,7 @@ class TextualInversionTrainer: ) # 指定ステップごとにモデルを保存 - if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and \ - args.save_every_n_steps_after_x is not None and global_step >= args.save_every_n_steps_after_x: - + if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and global_step >= args.save_every_n_steps_after_x: accelerator.wait_for_everyone() if accelerator.is_main_process: updated_embs_list = [] diff --git a/train_textual_inversion_XTI.py b/train_textual_inversion_XTI.py index 2ab48b63..1433c71d 100644 --- a/train_textual_inversion_XTI.py +++ b/train_textual_inversion_XTI.py @@ -499,9 +499,7 @@ def train(args): # ) # 指定ステップごとにモデルを保存 - if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and \ - args.save_every_n_steps_after_x is not None and global_step >= args.save_every_n_steps_after_x: - + if args.save_every_n_steps is not None and global_step % args.save_every_n_steps == 0 and global_step >= args.save_every_n_steps_after_x: accelerator.wait_for_everyone() if accelerator.is_main_process: updated_embs = (