diff options
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 0b26965..6de3a75 100644 --- a/train_lora.py +++ b/train_lora.py | |||
| @@ -927,7 +927,7 @@ def main(): | |||
| 927 | # LORA | 927 | # LORA |
| 928 | # -------------------------------------------------------------------------------- | 928 | # -------------------------------------------------------------------------------- |
| 929 | 929 | ||
| 930 | lora_output_dir = output_dir / "pti" | 930 | lora_output_dir = output_dir / "lora" |
| 931 | lora_checkpoint_output_dir = lora_output_dir / "model" | 931 | lora_checkpoint_output_dir = lora_output_dir / "model" |
| 932 | lora_sample_output_dir = lora_output_dir / "samples" | 932 | lora_sample_output_dir = lora_output_dir / "samples" |
| 933 | 933 | ||