diff options
author | Volpeon <git@volpeon.ink> | 2023-03-03 22:09:24 +0100 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-03-03 22:09:24 +0100 |
commit | 220806dbd21da3ba83c14096225c31824dfe81df (patch) | |
tree | a201272876bfe894f9d504d1582ac022add4b205 /train_dreambooth.py | |
parent | Implemented different noise offset (diff) | |
download | textual-inversion-diff-220806dbd21da3ba83c14096225c31824dfe81df.tar.gz textual-inversion-diff-220806dbd21da3ba83c14096225c31824dfe81df.tar.bz2 textual-inversion-diff-220806dbd21da3ba83c14096225c31824dfe81df.zip |
Removed offset noise from training, added init offset to pipeline
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py index 8571dff..9b91172 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
@@ -526,7 +526,6 @@ def main(): | |||
526 | with_prior_preservation=args.num_class_images != 0, | 526 | with_prior_preservation=args.num_class_images != 0, |
527 | prior_loss_weight=args.prior_loss_weight, | 527 | prior_loss_weight=args.prior_loss_weight, |
528 | no_val=args.valid_set_size == 0, | 528 | no_val=args.valid_set_size == 0, |
529 | # noise_offset=0, | ||
530 | ) | 529 | ) |
531 | 530 | ||
532 | checkpoint_output_dir = output_dir / "model" | 531 | checkpoint_output_dir = output_dir / "model" |