Commit
•
9b22310
1
Parent(s):
0ea5bec
Update README.md
Browse files
README.md
CHANGED
@@ -31,8 +31,13 @@ You are Dolphin, a helpful AI assistant.<|im_end|>
|
|
31 |
|
32 |
ChatML prompt template is available as a chat template, which means you can format messages using the tokenizer.apply_chat_template() method:
|
33 |
```
|
|
|
|
|
|
|
|
|
|
|
34 |
messages = [
|
35 |
-
{"role": "system", "content": "You are
|
36 |
{"role": "user", "content": "Hello BrokenLlama, what can you do for me?"}
|
37 |
]
|
38 |
|
|
|
31 |
|
32 |
ChatML prompt template is available as a chat template, which means you can format messages using the tokenizer.apply_chat_template() method:
|
33 |
```
|
34 |
+
from transformers import AutoConfig, AutoModel, AutoTokenizer
|
35 |
+
config = AutoConfig.from_pretrained("investbrainsorg/BrokenLlama-3-8b")
|
36 |
+
model = AutoModel.from_pretrained("investbrainsorg/BrokenLlama-3-8b")
|
37 |
+
tokenizer = AutoTokenizer.from_pretrained("investbrainsorg/BrokenLlama-3-8b")
|
38 |
+
|
39 |
messages = [
|
40 |
+
{"role": "system", "content": "You are BrokenLlama, a helpful AI assistant."},
|
41 |
{"role": "user", "content": "Hello BrokenLlama, what can you do for me?"}
|
42 |
]
|
43 |
|