Update README.md
Browse files
README.md
CHANGED
@@ -21,4 +21,5 @@ model = LlamaForCausalLM.from_pretrained("decapoda-research/llama-7b-hf",
|
|
21 |
load_in_8bit=True,
|
22 |
device_map="auto")
|
23 |
# Load the LoRA model
|
24 |
-
model = PeftModel.from_pretrained(model, peft_model_id)
|
|
|
|
21 |
load_in_8bit=True,
|
22 |
device_map="auto")
|
23 |
# Load the LoRA model
|
24 |
+
model = PeftModel.from_pretrained(model, peft_model_id)
|
25 |
+
```
|