MuntasirHossain commited on
Commit
ee1cea2
1 Parent(s): c29e4cf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -71,7 +71,7 @@ Daniel: I really hope Amiens is worth it!
71
  """
72
 
73
  input = tokenizer(text, return_tensors="pt")
74
- outputs = model.generate(input_ids=input["input_ids"].cuda(), max_new_tokens=40) # outputs = model.generate(input_ids=input["input_ids"].to('cuda'), max_new_tokens=50)
75
  print("Summary: ", tokenizer.decode(outputs[0], skip_special_tokens=True))
76
 
77
  Summary: Emma and Peter are going to Amiens tomorrow. Daniel will join them. They will meet at Emma's place by 9 am. They will not have any parties tonight.
 
71
  """
72
 
73
  input = tokenizer(text, return_tensors="pt")
74
+ outputs = model.generate(input_ids=input["input_ids"].cuda(), max_new_tokens=40)
75
  print("Summary: ", tokenizer.decode(outputs[0], skip_special_tokens=True))
76
 
77
  Summary: Emma and Peter are going to Amiens tomorrow. Daniel will join them. They will meet at Emma's place by 9 am. They will not have any parties tonight.