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 | ||