LiangJiang commited on
Commit
3bf9048
·
verified ·
1 Parent(s): 8053818

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -121,7 +121,7 @@ text = tokenizer.apply_chat_template(
121
  add_generation_prompt=True,
122
  enable_thinking=True
123
  )
124
- model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
125
 
126
  generated_ids = model.generate(
127
  **model_inputs,
 
121
  add_generation_prompt=True,
122
  enable_thinking=True
123
  )
124
+ model_inputs = tokenizer([text], return_tensors="pt", return_token_type_ids=False).to(model.device)
125
 
126
  generated_ids = model.generate(
127
  **model_inputs,