Update README.md
Browse files
README.md
CHANGED
@@ -92,7 +92,7 @@ messages = [
|
|
92 |
{"role": "system", "content": "You are Hermes 2."},
|
93 |
{"role": "user", "content": "Hello, who are you?"}
|
94 |
]
|
95 |
-
gen_input = tokenizer.apply_chat_template(
|
96 |
model.generate(**gen_input)
|
97 |
```
|
98 |
|
|
|
92 |
{"role": "system", "content": "You are Hermes 2."},
|
93 |
{"role": "user", "content": "Hello, who are you?"}
|
94 |
]
|
95 |
+
gen_input = tokenizer.apply_chat_template(messages, return_tensors="pt")
|
96 |
model.generate(**gen_input)
|
97 |
```
|
98 |
|