diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-02 17:12:06 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-02 17:12:06 +0100 |
| commit | b1d7b2962e6454f8e72bd64efe08dc80a1f2d3aa (patch) | |
| tree | 8e93d7c884b2ba0d86748ad200173fcdb39df5ef /train_dreambooth.py | |
| parent | Save args before training, too (diff) | |
| download | textual-inversion-diff-b1d7b2962e6454f8e72bd64efe08dc80a1f2d3aa.tar.gz textual-inversion-diff-b1d7b2962e6454f8e72bd64efe08dc80a1f2d3aa.tar.bz2 textual-inversion-diff-b1d7b2962e6454f8e72bd64efe08dc80a1f2d3aa.zip | |
Fix
Diffstat (limited to 'train_dreambooth.py')
| -rw-r--r-- | train_dreambooth.py | 4 |
1 files changed, 0 insertions, 4 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py index cd0bf67..05f6cb5 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
| @@ -955,10 +955,6 @@ def main(): | |||
| 955 | ) | 955 | ) |
| 956 | global_progress_bar.set_description("Total progress") | 956 | global_progress_bar.set_description("Total progress") |
| 957 | 957 | ||
| 958 | save_args(basepath, args, { | ||
| 959 | "global_step": global_step + global_step_offset | ||
| 960 | }) | ||
| 961 | |||
| 962 | try: | 958 | try: |
| 963 | for epoch in range(num_epochs): | 959 | for epoch in range(num_epochs): |
| 964 | if accelerator.is_main_process: | 960 | if accelerator.is_main_process: |