diff options
| author | Volpeon <git@volpeon.ink> | 2023-02-15 15:11:37 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-02-15 15:11:37 +0100 |
| commit | 0b8568ebfb484b9155d19c567e890ff52776544d (patch) | |
| tree | 09ff3256ca8d2c9359a236e7bdb274f539c5253e /training | |
| parent | Improved batch padding (diff) | |
| download | textual-inversion-diff-0b8568ebfb484b9155d19c567e890ff52776544d.tar.gz textual-inversion-diff-0b8568ebfb484b9155d19c567e890ff52776544d.tar.bz2 textual-inversion-diff-0b8568ebfb484b9155d19c567e890ff52776544d.zip | |
Update
Diffstat (limited to 'training')
| -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 a9c7a8a..e1035ce 100644 --- a/training/functional.py +++ b/training/functional.py | |||
| @@ -556,7 +556,7 @@ def train( | |||
| 556 | global_step_offset: int = 0, | 556 | global_step_offset: int = 0, |
| 557 | with_prior_preservation: bool = False, | 557 | with_prior_preservation: bool = False, |
| 558 | prior_loss_weight: float = 1.0, | 558 | prior_loss_weight: float = 1.0, |
| 559 | low_freq_noise: float = 0.05, | 559 | low_freq_noise: float = 0.1, |
| 560 | **kwargs, | 560 | **kwargs, |
| 561 | ): | 561 | ): |
| 562 | text_encoder, unet, optimizer, train_dataloader, val_dataloader, lr_scheduler, extra = strategy.prepare( | 562 | text_encoder, unet, optimizer, train_dataloader, val_dataloader, lr_scheduler, extra = strategy.prepare( |