avacaondata commited on
Commit
e5691dc
1 Parent(s): 05cc9e7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -74,7 +74,7 @@ messages = [
74
  ]
75
  input_ids = tokenizer.apply_chat_template(messages, tokenize=True, return_tensors="pt")
76
  with torch.no_grad():
77
- output = model.generate(**input_ids, max_length=50)
78
  print(tokenizer.decode(output[0], skip_special_tokens=True))
79
  ```
80
 
 
74
  ]
75
  input_ids = tokenizer.apply_chat_template(messages, tokenize=True, return_tensors="pt")
76
  with torch.no_grad():
77
+ output = model.generate(input_ids, max_new_tokens=50)
78
  print(tokenizer.decode(output[0], skip_special_tokens=True))
79
  ```
80