mlabonne zwischenraum commited on
Commit
7728373
·
1 Parent(s): 7081022

Update README.md (#5)

Browse files

- Update README.md (120bf032c36dd1273de754669da040dba7ad4982)


Co-authored-by: Jens Lücke <zwischenraum@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -155,7 +155,7 @@ input_ids = tokenizer.apply_chat_template(
155
  add_generation_prompt=True,
156
  return_tensors="pt",
157
  tokenize=True,
158
- ).to(model.device)
159
 
160
  output = model.generate(
161
  input_ids,
 
155
  add_generation_prompt=True,
156
  return_tensors="pt",
157
  tokenize=True,
158
+ )["input_ids"].to(model.device)
159
 
160
  output = model.generate(
161
  input_ids,