From f5b86b44565aaaa92543989a85ea5d88ca9b1c0c Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 7 Apr 2023 11:02:47 +0200 Subject: Fix --- 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 0b26965..6de3a75 100644 --- a/train_lora.py +++ b/train_lora.py @@ -927,7 +927,7 @@ def main(): # LORA # -------------------------------------------------------------------------------- - lora_output_dir = output_dir / "pti" + lora_output_dir = output_dir / "lora" lora_checkpoint_output_dir = lora_output_dir / "model" lora_sample_output_dir = lora_output_dir / "samples" -- cgit v1.2.3-54-g00ecf