Implemented different noise offset

1 files changed, 2 insertions, 2 deletions

diff --git a/train_dreambooth.py b/train_dreambooth.py
index 6d699f3..8571dff 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -438,7 +438,7 @@ def main():
 
     accelerator = Accelerator(
         log_with=LoggerType.TENSORBOARD,
-        logging_dir=f"{output_dir}",
+        project_dir=f"{output_dir}",
         mixed_precision=args.mixed_precision
     )
 
@@ -526,7 +526,7 @@ def main():
         with_prior_preservation=args.num_class_images != 0,
         prior_loss_weight=args.prior_loss_weight,
         no_val=args.valid_set_size == 0,
-        # low_freq_noise=0,
+        # noise_offset=0,
     )
 
     checkpoint_output_dir = output_dir / "model"