eunhwanpark-motiftech commited on
Commit
48ae3c6
·
verified ·
1 Parent(s): 8d10c7c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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': 'system', 'content': query},
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',