Update README.md
Browse files
README.md
CHANGED
@@ -41,7 +41,7 @@ model = LlamaForCausalLM.from_pretrained(base)
|
|
41 |
model = PeftModel.from_pretrained(model, finetuned).to("cuda")
|
42 |
```
|
43 |
|
44 |
-
For output generation use the following code
|
45 |
|
46 |
|
47 |
```python
|
|
|
41 |
model = PeftModel.from_pretrained(model, finetuned).to("cuda")
|
42 |
```
|
43 |
|
44 |
+
For output generation use the following code. Feel free to adjust the parameters according to your needs.
|
45 |
|
46 |
|
47 |
```python
|