Implemented different noise offset

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():
 
     accelerator = Accelerator(
         log_with=LoggerType.TENSORBOARD,
-        logging_dir=f"{output_dir}",
+        project_dir=f"{output_dir}",
         mixed_precision=args.mixed_precision
     )
 
@@ -607,7 +607,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,
         strategy=textual_inversion_strategy,
         num_train_epochs=args.num_train_epochs,
         gradient_accumulation_steps=args.gradient_accumulation_steps,