Update README.md
Browse files
README.md
CHANGED
@@ -5,6 +5,10 @@ datasets:
|
|
5 |
language:
|
6 |
- en
|
7 |
---
|
|
|
|
|
|
|
|
|
8 |
from peft import PeftModel, PeftConfig
|
9 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
10 |
|
@@ -37,4 +41,5 @@ response = tokenizer.decode(response[0],
|
|
37 |
skip_special_tokens=False,
|
38 |
clean_up_tokenization_spaces=False
|
39 |
)
|
40 |
-
print(response)
|
|
|
|
5 |
language:
|
6 |
- en
|
7 |
---
|
8 |
+
|
9 |
+
## Inference Example:
|
10 |
+
|
11 |
+
```python
|
12 |
from peft import PeftModel, PeftConfig
|
13 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
14 |
|
|
|
41 |
skip_special_tokens=False,
|
42 |
clean_up_tokenization_spaces=False
|
43 |
)
|
44 |
+
print(response)
|
45 |
+
```
|