fix lora tests

Lightning-AI · Mar 8, 2024 · 2e918d5 · 2e918d5
1 parent f483d68
commit 2e918d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/litgpt/finetune/lora.py b/litgpt/finetune/lora.py
@@ -264,7 +264,7 @@ def fit(
             fabric.barrier()
 
         if train.save_interval is not None and not is_accumulating and step_count % train.save_interval == 0:
-            checkpoint_file = out_dir / f"step-{step_count:06d}" / "lit_model.pth"
+            checkpoint_file = out_dir / f"step-{step_count:06d}" / "lit_model.pth.lora"
             checkpoint_file.parent.mkdir(parents=True, exist_ok=True)
             save_lora_checkpoint(fabric, model, checkpoint_file)
             if fabric.global_rank == 0: