Update merge-hf-and-lora-to-hf.py
Co-authored-by: slaren <slarengh@gmail.com>
This commit is contained in:
parent
b6a30489a7
commit
39287b06da
1 changed files with 1 additions and 2 deletions
|
@ -5,8 +5,7 @@ from peft import PeftModel
|
|||
|
||||
# args with description.
|
||||
parser = argparse.ArgumentParser(
|
||||
prog="Merge HF file with Lora\n",
|
||||
description="Please locate HF format model path with pytorch_*.bin inside, lora path with adapter_config.json and adapter_model.bin.",
|
||||
description="Merge HF model with LoRA",
|
||||
)
|
||||
|
||||
# The original base model checkpoint dir
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue