Commit
•
51418ff
1
Parent(s):
0c3ee37
Update README.md
Browse files
README.md
CHANGED
|
@@ -26,9 +26,11 @@ Use the code below to get started with the model.
|
|
| 26 |
|
| 27 |
```python
|
| 28 |
from peft import AutoPeftModelForCausalLM
|
|
|
|
| 29 |
|
| 30 |
model = AutoPeftModelForCausalLM.from_pretrained("baris-yazici/liar_stabilityai_stablelm-2-zephyr-1_6b_PROMPT_TUNING_CAUSAL_LM").to("cuda")
|
| 31 |
tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-2-zephyr-1_6b")
|
|
|
|
| 32 |
```
|
| 33 |
|
| 34 |
## Training Details
|
|
|
|
| 26 |
|
| 27 |
```python
|
| 28 |
from peft import AutoPeftModelForCausalLM
|
| 29 |
+
from transformers import AutoTokenizer
|
| 30 |
|
| 31 |
model = AutoPeftModelForCausalLM.from_pretrained("baris-yazici/liar_stabilityai_stablelm-2-zephyr-1_6b_PROMPT_TUNING_CAUSAL_LM").to("cuda")
|
| 32 |
tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-2-zephyr-1_6b")
|
| 33 |
+
|
| 34 |
```
|
| 35 |
|
| 36 |
## Training Details
|