From 1fd5d10b07d65d287fc5a7a8c9f0ed070aba7449 Mon Sep 17 00:00:00 2001 From: FNsi <125447286+FNsi@users.noreply.github.com> Date: Mon, 22 May 2023 19:27:45 +0800 Subject: [PATCH] Update merge-hf-and-lora-to-hf.py Co-authored-by: slaren --- merge-hf-and-lora-to-hf.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/merge-hf-and-lora-to-hf.py b/merge-hf-and-lora-to-hf.py index 2dcd7c1ba..ae136b092 100644 --- a/merge-hf-and-lora-to-hf.py +++ b/merge-hf-and-lora-to-hf.py @@ -10,23 +10,23 @@ parser = argparse.ArgumentParser( # The original base model checkpoint dir parser.add_argument( - "--model_path", + "--model", type=str, default="decapoda-research/llama-7b-hf", - help="Directory contain original HF model", + help="HF model name or path", ) # The finetuned lora model checkpoint dir parser.add_argument( - "--lora_path", + "--lora", type=str, - default="decapoda-research/lora", - help="Directory contain Lora", + required=True, + help="LoRA model name or path", ) # The output dir parser.add_argument( - "--out_path", + "--out", type=str, - default="decapoda-research/lora-merged", + required=True, help="Directory store merged HF model", )