From 220806dbd21da3ba83c14096225c31824dfe81df Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 3 Mar 2023 22:09:24 +0100 Subject: Removed offset noise from training, added init offset to pipeline --- train_ti.py | 1 - 1 file changed, 1 deletion(-) (limited to 'train_ti.py') diff --git a/train_ti.py b/train_ti.py index bc9348d..c139cc0 100644 --- a/train_ti.py +++ b/train_ti.py @@ -607,7 +607,6 @@ def main(): with_prior_preservation=args.num_class_images != 0, prior_loss_weight=args.prior_loss_weight, no_val=args.valid_set_size == 0, - noise_offset=0, strategy=textual_inversion_strategy, num_train_epochs=args.num_train_epochs, gradient_accumulation_steps=args.gradient_accumulation_steps, -- cgit v1.2.3-54-g00ecf