Skip to content

Commit

Permalink
Update finetune.py
Browse files Browse the repository at this point in the history
  • Loading branch information
AkariAsai authored Mar 19, 2024
1 parent 141dd48 commit b37103d
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions retrieval_lm/finetune.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,11 +99,6 @@ def parse_args():
default=0.1,
help="The dropout rate of lora modules.",
)
parser.add_argument(
"--save_merged_lora_model",
action="store_true",
help="If passed, will merge the lora modules and save the entire model.",
)
parser.add_argument(
"--use_flash_attn",
action="store_true",
Expand Down

0 comments on commit b37103d

Please sign in to comment.