katuni4ka commited on
Commit
e313eec
1 Parent(s): 255a3b9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -39,7 +39,7 @@ messages = [
39
  {"role": "user", "content": "Do you have mayonnaise recipes?"}
40
  ]
41
 
42
- inputs = tokenizer.apply_chat_template(messages, return_tensors="pt").to("cuda")
43
 
44
  outputs = model.generate(inputs, max_new_tokens=20)
45
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 
39
  {"role": "user", "content": "Do you have mayonnaise recipes?"}
40
  ]
41
 
42
+ inputs = tokenizer.apply_chat_template(messages, return_tensors="pt")
43
 
44
  outputs = model.generate(inputs, max_new_tokens=20)
45
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))