AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Fix --resume_from_checkpoint step in train_text_to_image.py (#1914)

Browse Source 
fix resume step in train_text_to_image example
This commit is contained in:
Manfred Lindmark
2023-01-04 13:05:55 +01:00
committed by GitHub
parent 675ef1ffbd
commit f7154f859c
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/text_to_image/train_text_to_image.py
View File

@@ -685,9 +685,8 @@ def main():
accelerator.load_state(os.path.join(args.output_dir, path))
global_step = int(path.split("-")[1])
resume_global_step = global_step * args.gradient_accumulation_steps
first_epoch = resume_global_step // num_update_steps_per_epoch
resume_step = resume_global_step % num_update_steps_per_epoch
first_epoch = global_step // num_update_steps_per_epoch
resume_step = global_step % num_update_steps_per_epoch
# Only show the progress bar once on each machine.
progress_bar = tqdm(range(global_step, args.max_train_steps), disable=not accelerator.is_local_main_process)