yitongy commited on
Commit
872c7c0
·
verified ·
1 Parent(s): f2d033f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -75,7 +75,7 @@ messages = [{"role": "user", "content": prompt}]
75
  text = tokenizer.apply_chat_template(messages,
76
  tokenize=False,
77
  add_generation_prompt=True
78
- >>> )
79
  model_inputs = tokenizer(text, return_tensors="pt").to(model.device)
80
  generated_ids = model.generate(
81
  **model_inputs,
 
75
  text = tokenizer.apply_chat_template(messages,
76
  tokenize=False,
77
  add_generation_prompt=True
78
+ )
79
  model_inputs = tokenizer(text, return_tensors="pt").to(model.device)
80
  generated_ids = model.generate(
81
  **model_inputs,