Update README.md
Browse files
README.md
CHANGED
|
@@ -134,7 +134,7 @@ def main(args):
|
|
| 134 |
{"role": "assistant", "content": "I'm doing great. How can I help you today?"},
|
| 135 |
{"role": "user", "content": "I'd like to show off how chat templating works!"},
|
| 136 |
]
|
| 137 |
-
input_ids = tokenizer.apply_chat_template(chat,
|
| 138 |
input_ids = input_ids.to(device=DEVICE)
|
| 139 |
|
| 140 |
# Please adjust parameters like top_p appropriately for your use case.
|
|
|
|
| 134 |
{"role": "assistant", "content": "I'm doing great. How can I help you today?"},
|
| 135 |
{"role": "user", "content": "I'd like to show off how chat templating works!"},
|
| 136 |
]
|
| 137 |
+
input_ids = tokenizer.apply_chat_template(chat, return_tensors="pt", tokenize=True)
|
| 138 |
input_ids = input_ids.to(device=DEVICE)
|
| 139 |
|
| 140 |
# Please adjust parameters like top_p appropriately for your use case.
|