merge-hf-and-lora-to-hf.py
This commit is contained in:
parent
28bec1eb25
commit
5dbdc65700
1 changed files with 0 additions and 1 deletions
|
@ -48,7 +48,6 @@ model = LlamaForCausalLM.from_pretrained(
|
||||||
device_map={"": "cpu"},
|
device_map={"": "cpu"},
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
# peft loaded. load lora.
|
# peft loaded. load lora.
|
||||||
model = PeftModel.from_pretrained(
|
model = PeftModel.from_pretrained(
|
||||||
model,
|
model,
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue