From 602c5ff86ce5d9b8aee545dd243ff04d8bddf405 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sun, 9 Apr 2023 10:13:02 +0200 Subject: Fix --- training/functional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'training') diff --git a/training/functional.py b/training/functional.py index 71b2fe9..7d49782 100644 --- a/training/functional.py +++ b/training/functional.py @@ -606,8 +606,8 @@ def train_loop( # Create the pipeline using using the trained modules and save it. if accelerator.is_main_process: print("Finished!") - on_checkpoint(global_step + global_step_offset, "end") on_sample(global_step + global_step_offset) + on_checkpoint(global_step + global_step_offset, "end") except KeyboardInterrupt: if accelerator.is_main_process: -- cgit v1.2.3-70-g09d2