Humachine commited on
Commit
f86fbdd
·
verified ·
1 Parent(s): 4bf3e50

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -116,9 +116,9 @@ messages = [
116
 
117
  ### Generate Output
118
  ```python
119
- inputs = tokenizer.apply_chat_template(
120
- messages, add_generation_prompt=True, return_dict=True, return_tensors="pt"
121
- ).to("cuda",dtype=torch.bfloat16)
122
 
123
  output_tokens = model.generate(**inputs, max_new_tokens=2048, use_cache=True, temperature=0.2)
124
  output_text = tokenizer.decode(output_tokens[0], skip_special_tokens=True)
 
116
 
117
  ### Generate Output
118
  ```python
119
+ text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
120
+
121
+ inputs = tokenizer(text, return_tensors="pt").to("cuda",dtype=torch.bfloat16)
122
 
123
  output_tokens = model.generate(**inputs, max_new_tokens=2048, use_cache=True, temperature=0.2)
124
  output_text = tokenizer.decode(output_tokens[0], skip_special_tokens=True)