macadeliccc
commited on
Commit
•
8dbceaa
1
Parent(s):
4159bb3
Update README.md
Browse files
README.md
CHANGED
@@ -57,5 +57,15 @@ dtype: bfloat16
|
|
57 |
TODO
|
58 |
|
59 |
## Code Example
|
|
|
|
|
60 |
|
61 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
57 |
TODO
|
58 |
|
59 |
## Code Example
|
60 |
+
```python
|
61 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
62 |
|
63 |
+
tokenizer = AutoTokenizer.from_pretrained("macadeliccc/samantha-1.1-MBX-7B")
|
64 |
+
model = AutoModelForCausalLM.from_pretrained("macadeliccc/samanth-1.1-MBX-7B")
|
65 |
+
|
66 |
+
messages = [
|
67 |
+
{"role": "system", "content": "You are a caring and empathetic sentient AI companion named Samantha."},
|
68 |
+
{"role": "user", "content": "Hello, what is your name?"}
|
69 |
+
]
|
70 |
+
gen_input = tokenizer.apply_chat_template(messages, return_tensors="pt")
|
71 |
+
```
|