From e32b4d4c04a31b22051740e5f26e16960464f787 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 3 Mar 2023 18:53:15 +0100 Subject: Implemented different noise offset --- train_lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_lora.py') diff --git a/train_lora.py b/train_lora.py index 0a3d4c9..e213e3d 100644 --- a/train_lora.py +++ b/train_lora.py @@ -398,7 +398,7 @@ def main(): accelerator = Accelerator( log_with=LoggerType.TENSORBOARD, - logging_dir=f"{output_dir}", + project_dir=f"{output_dir}", mixed_precision=args.mixed_precision ) -- cgit v1.2.3-70-g09d2