Update README.md
Browse files
README.md
CHANGED
|
@@ -26,6 +26,7 @@ model = LlamaForCausalLM.from_pretrained("decapoda-research/llama-7b-hf",
|
|
| 26 |
model = PeftModel.from_pretrained(model, peft_model_id)
|
| 27 |
```
|
| 28 |
## Prompt Template
|
|
|
|
| 29 |
|
| 30 |
```python
|
| 31 |
def generate_prompt(instruction, input=None):
|
|
|
|
| 26 |
model = PeftModel.from_pretrained(model, peft_model_id)
|
| 27 |
```
|
| 28 |
## Prompt Template
|
| 29 |
+
Prepare the prompt template
|
| 30 |
|
| 31 |
```python
|
| 32 |
def generate_prompt(instruction, input=None):
|