diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-11 21:57:43 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-11 21:57:43 +0100 |
| commit | d12634cb211e10223ad77a43a562d80678b2d662 (patch) | |
| tree | 47b0678a33617e49090ae6690c8b83bc867d3b34 /train_ti.py | |
| parent | TI: Use grad clipping from LoRA #104 (diff) | |
| download | textual-inversion-diff-d12634cb211e10223ad77a43a562d80678b2d662.tar.gz textual-inversion-diff-d12634cb211e10223ad77a43a562d80678b2d662.tar.bz2 textual-inversion-diff-d12634cb211e10223ad77a43a562d80678b2d662.zip | |
Heck
Diffstat (limited to 'train_ti.py')
| -rw-r--r-- | train_ti.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py index 1054a5d..e6c437e 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -974,7 +974,7 @@ def main(): | |||
| 974 | 974 | ||
| 975 | try: | 975 | try: |
| 976 | for epoch in range(num_epochs): | 976 | for epoch in range(num_epochs): |
| 977 | if accelerator.is_main_process and False: | 977 | if accelerator.is_main_process: |
| 978 | if epoch % args.sample_frequency == 0: | 978 | if epoch % args.sample_frequency == 0: |
| 979 | checkpointer.save_samples(global_step + global_step_offset, args.sample_steps) | 979 | checkpointer.save_samples(global_step + global_step_offset, args.sample_steps) |
| 980 | 980 | ||