[Test] Save local model path in PEFT adapter config

pytorch · Nov 19, 2024 · c69a4f8 · c69a4f8
1 parent bce7091
commit c69a4f8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/recipes/lora_finetune_single_device.py b/recipes/lora_finetune_single_device.py
@@ -619,6 +619,7 @@ def save_checkpoint(self, epoch: int) -> None:
                 self._apply_lora_to_output,
             ),
             "peft_type": "LORA",
+            "base_model_name_or_path": str(self._checkpointer._checkpoint_dir),
         }
         ckpt_dict.update({training.ADAPTER_CONFIG: adapter_config})