diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-16 10:03:05 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-16 10:03:05 +0100 |
| commit | d5696615a84a768307e82d13e50b4aef64f69dbd (patch) | |
| tree | 47cfaa5b8922edbfd567739ecd770977e339f8d7 /training | |
| parent | Implemented extended Dreambooth training (diff) | |
| download | textual-inversion-diff-d5696615a84a768307e82d13e50b4aef64f69dbd.tar.gz textual-inversion-diff-d5696615a84a768307e82d13e50b4aef64f69dbd.tar.bz2 textual-inversion-diff-d5696615a84a768307e82d13e50b4aef64f69dbd.zip | |
Extended Dreambooth: Train TI tokens separately
Diffstat (limited to 'training')
| -rw-r--r-- | training/functional.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/training/functional.py b/training/functional.py index f5c111e..1b6162b 100644 --- a/training/functional.py +++ b/training/functional.py | |||
| @@ -486,6 +486,7 @@ def train_loop( | |||
| 486 | if accelerator.is_main_process: | 486 | if accelerator.is_main_process: |
| 487 | print("Interrupted") | 487 | print("Interrupted") |
| 488 | on_checkpoint(global_step + global_step_offset, "end") | 488 | on_checkpoint(global_step + global_step_offset, "end") |
| 489 | raise KeyboardInterrupt | ||
| 489 | 490 | ||
| 490 | 491 | ||
| 491 | def train( | 492 | def train( |