DongHyunKim commited on
Commit
1f67d50
·
verified ·
1 Parent(s): 4ba3236

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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, chat_template="llm", 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.
 
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.