Update README.md
Browse files
README.md
CHANGED
@@ -217,7 +217,7 @@ tokenizer = AutoTokenizer.from_pretrained(
|
|
217 |
input_ids = tokenizer.apply_chat_template(
|
218 |
[
|
219 |
{'role': 'system', 'content': 'you are an helpful assistant'},
|
220 |
-
{'role': '
|
221 |
],
|
222 |
add_generation_prompt = True,
|
223 |
return_tensors='pt',
|
|
|
217 |
input_ids = tokenizer.apply_chat_template(
|
218 |
[
|
219 |
{'role': 'system', 'content': 'you are an helpful assistant'},
|
220 |
+
{'role': 'user', 'content': query},
|
221 |
],
|
222 |
add_generation_prompt = True,
|
223 |
return_tensors='pt',
|