georgethrax commited on
Commit
de0bfbe
·
verified ·
1 Parent(s): ae28fc1

fix: update Transformers quickstart to be compatible with transformers v5.x

Browse files
Files changed (1) hide show
  1. README-cn.md +3 -2
README-cn.md CHANGED
@@ -169,11 +169,12 @@ inputs = tokenizer.apply_chat_template(
169
  tokenize=True,
170
  add_generation_prompt=True,
171
  enable_thinking=False,
 
172
  return_tensors="pt",
173
  ).to(model.device)
174
 
175
- outputs = model.generate(inputs, max_new_tokens=128)
176
- print(tokenizer.decode(outputs[0][inputs.shape[-1]:], skip_special_tokens=True))
177
  ```
178
 
179
  推荐的 chat template 采样参数:
 
169
  tokenize=True,
170
  add_generation_prompt=True,
171
  enable_thinking=False,
172
+ return_dict=True,
173
  return_tensors="pt",
174
  ).to(model.device)
175
 
176
+ outputs = model.generate(**inputs, max_new_tokens=128)
177
+ print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:], skip_special_tokens=True))
178
  ```
179
 
180
  推荐的 chat template 采样参数: