vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update merge-hf-and-lora-to-hf.py

Browse source 
Co-authored-by: slaren <slarengh@gmail.com>
This commit is contained in:
FNsi 2023-05-22 19:26:00 +08:00 • committed by GitHub
parent b6a30489a7
commit 39287b06da
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
merge-hf-and-lora-to-hf.py
View file

@ -5,8 +5,7 @@ from peft import PeftModel
# args with description. # args with description.
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
prog="Merge HF file with Lora\n", description="Merge HF model with LoRA",
description="Please locate HF format model path with pytorch_*.bin inside, lora path with adapter_config.json and adapter_model.bin.",
) )
# The original base model checkpoint dir # The original base model checkpoint dir