Skip to content

Commit

Permalink
Update merge_lora.py docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
carmocca authored and rasbt committed Apr 3, 2024
1 parent cae9871 commit 98b288d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion litgpt/scripts/merge_lora.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ def merge_lora(
Args:
checkpoint_dir: Path to the checkpoint directory with trained LoRA weights, which is the output of
``litgpt finetune --method lora``.
``litgpt finetune lora``.
pretrained_checkpoint_dir: Optional path to the checkpoint directory with the weights of the base model
corresponding to the LoRA checkpoint. By default, this will automatically be inferred from the metadata
in the given `checkpoint_dir` directory. Only set this if the base model's checkpoint directory
Expand Down

0 comments on commit 98b288d

Please sign in to comment.