Update README.md

README.md

@@ -69,6 +69,9 @@ text = tokenizer.apply_chat_template(
 
 inputs = tokenizer(text, return_tensors="pt").to(model.device)
 
+if 'token_type_ids' in inputs:
+    inputs.pop('token_type_ids')
+
 outputs = model.generate(
     **inputs,
     max_new_tokens=200,