diff options
| author | Volpeon <git@volpeon.ink> | 2023-04-09 10:13:02 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-04-09 10:13:02 +0200 |
| commit | 602c5ff86ce5d9b8aee545dd243ff04d8bddf405 (patch) | |
| tree | 62e2d15844ac8c2e9c83801c79976efb8d152f72 /training/functional.py | |
| parent | Made Lora script interactive (diff) | |
| download | textual-inversion-diff-602c5ff86ce5d9b8aee545dd243ff04d8bddf405.tar.gz textual-inversion-diff-602c5ff86ce5d9b8aee545dd243ff04d8bddf405.tar.bz2 textual-inversion-diff-602c5ff86ce5d9b8aee545dd243ff04d8bddf405.zip | |
Fix
Diffstat (limited to 'training/functional.py')
| -rw-r--r-- | training/functional.py | 2 |
1 files changed, 1 insertions, 1 deletions
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( | |||
| 606 | # Create the pipeline using using the trained modules and save it. | 606 | # Create the pipeline using using the trained modules and save it. |
| 607 | if accelerator.is_main_process: | 607 | if accelerator.is_main_process: |
| 608 | print("Finished!") | 608 | print("Finished!") |
| 609 | on_checkpoint(global_step + global_step_offset, "end") | ||
| 610 | on_sample(global_step + global_step_offset) | 609 | on_sample(global_step + global_step_offset) |
| 610 | on_checkpoint(global_step + global_step_offset, "end") | ||
| 611 | 611 | ||
| 612 | except KeyboardInterrupt: | 612 | except KeyboardInterrupt: |
| 613 | if accelerator.is_main_process: | 613 | if accelerator.is_main_process: |