diff options
| author | Volpeon <git@volpeon.ink> | 2023-03-03 18:53:15 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-03-03 18:53:15 +0100 |
| commit | e32b4d4c04a31b22051740e5f26e16960464f787 (patch) | |
| tree | 9fc842453e4974b936e64b4f012acfef726a8e51 /train_lora.py | |
| parent | Low freq noise with randomized strength (diff) | |
| download | textual-inversion-diff-e32b4d4c04a31b22051740e5f26e16960464f787.tar.gz textual-inversion-diff-e32b4d4c04a31b22051740e5f26e16960464f787.tar.bz2 textual-inversion-diff-e32b4d4c04a31b22051740e5f26e16960464f787.zip | |
Implemented different noise offset
Diffstat (limited to 'train_lora.py')
| -rw-r--r-- | train_lora.py | 2 |
1 files changed, 1 insertions, 1 deletions
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(): | |||
| 398 | 398 | ||
| 399 | accelerator = Accelerator( | 399 | accelerator = Accelerator( |
| 400 | log_with=LoggerType.TENSORBOARD, | 400 | log_with=LoggerType.TENSORBOARD, |
| 401 | logging_dir=f"{output_dir}", | 401 | project_dir=f"{output_dir}", |
| 402 | mixed_precision=args.mixed_precision | 402 | mixed_precision=args.mixed_precision |
| 403 | ) | 403 | ) |
| 404 | 404 | ||