Skip to content

Commit

Permalink
Update examples/safelora/README.md
Browse files Browse the repository at this point in the history
Co-authored-by: Benjamin Bossan <[email protected]>
  • Loading branch information
chiayi-hsu and BenjaminBossan authored Nov 25, 2024
1 parent 71e9467 commit 9b0b06e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/safelora/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,6 @@ Next, you can load the base model of the Peft Model along with the Peft model it
from transformers import AutoModelForCausalLM
from peft import PeftConfig, PeftModel

model = AutoModelForCausalLM.from_pretrained('Your base model of the Peft model PATH')
model = PeftModel.from_pretrained(model, "Your SafeLoRA Weight PATH")
model = AutoModelForCausalLM.from_pretrained(<base-model-id>)
model = PeftModel.from_pretrained(model, <SafeLoRA-path>)
```

0 comments on commit 9b0b06e

Please sign in to comment.