Merge pull request #240 from NatalieHanzhina/main

Accelerator issue is fixed

Author: ShivamShrirao

examples/dreambooth/train_dreambooth.py

@@ -426,7 +426,7 @@ def main(args):
         gradient_accumulation_steps=args.gradient_accumulation_steps,
         mixed_precision=args.mixed_precision,
         log_with="tensorboard",
-        logging_dir=logging_dir,
+        project_dir=logging_dir,
     )
 
     logging.basicConfig(
@@ -866,4 +866,4 @@ def save_weights(step):
 
 if __name__ == "__main__":
     args = parse_args()
-    main(args)
+    main(args)