diff options
Diffstat (limited to 'train_ti.py')
-rw-r--r-- | train_ti.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/train_ti.py b/train_ti.py index 394711f..bc9348d 100644 --- a/train_ti.py +++ b/train_ti.py | |||
@@ -517,7 +517,7 @@ def main(): | |||
517 | 517 | ||
518 | accelerator = Accelerator( | 518 | accelerator = Accelerator( |
519 | log_with=LoggerType.TENSORBOARD, | 519 | log_with=LoggerType.TENSORBOARD, |
520 | logging_dir=f"{output_dir}", | 520 | project_dir=f"{output_dir}", |
521 | mixed_precision=args.mixed_precision | 521 | mixed_precision=args.mixed_precision |
522 | ) | 522 | ) |
523 | 523 | ||
@@ -607,7 +607,7 @@ def main(): | |||
607 | with_prior_preservation=args.num_class_images != 0, | 607 | with_prior_preservation=args.num_class_images != 0, |
608 | prior_loss_weight=args.prior_loss_weight, | 608 | prior_loss_weight=args.prior_loss_weight, |
609 | no_val=args.valid_set_size == 0, | 609 | no_val=args.valid_set_size == 0, |
610 | # low_freq_noise=0, | 610 | noise_offset=0, |
611 | strategy=textual_inversion_strategy, | 611 | strategy=textual_inversion_strategy, |
612 | num_train_epochs=args.num_train_epochs, | 612 | num_train_epochs=args.num_train_epochs, |
613 | gradient_accumulation_steps=args.gradient_accumulation_steps, | 613 | gradient_accumulation_steps=args.gradient_accumulation_steps, |