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

This commit is contained in:
FNsi 2023-05-21 11:19:18 +08:00 committed by GitHub
parent 28bec1eb25
commit 5dbdc65700
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -48,7 +48,6 @@ model = LlamaForCausalLM.from_pretrained(
device_map={"": "cpu"},
)
# peft loaded. load lora.
model = PeftModel.from_pretrained(
model,