From d892edcf7dedcfda72f068c2b3ce4fbc5d582d76 Mon Sep 17 00:00:00 2001
From: FNsi <125447286+FNsi@users.noreply.github.com>
Date: Sun, 21 May 2023 11:14:52 +0800
Subject: [PATCH] Update merge-hf-and-lora-to-hf.py

---
 merge-hf-and-lora-to-hf.py | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/merge-hf-and-lora-to-hf.py b/merge-hf-and-lora-to-hf.py
index bd3db4609..a8f2a0aff 100644
--- a/merge-hf-and-lora-to-hf.py
+++ b/merge-hf-and-lora-to-hf.py
@@ -11,11 +11,26 @@ parser = argparse.ArgumentParser(
 )
 
 # The original base model checkpoint dir
-parser.add_argument("--model_path", type=str, default="decapoda-research/llama-7b-hf", help="Directory contain original HF model")
+parser.add_argument(
+    "--model_path",
+    type=str,
+    default="decapoda-research/llama-7b-hf",
+    help="Directory contain original HF model",
+)
 # The finetuned lora model checkpoint dir
-parser.add_argument("--lora_path", type=str, default="decapoda-research/lora", help="Directory contain Lora ")
+parser.add_argument(
+    "--lora_path",
+    type=str,
+    default="decapoda-research/lora",
+    help="Directory contain Lora ",
+)
 # The output dir
-parser.add_argument("--out_path", type=str, default="decapoda-research/lora-merged", help="Directory  store merged HF model")
+parser.add_argument(
+    "--out_path",
+    type=str,
+    default="decapoda-research/lora-merged",
+    help="Directory  store merged HF model",
+)
 
 args = parser.parse_args()