rooa commited on
Commit
a94e500
1 Parent(s): d549dcc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -48,7 +48,7 @@ article = "" # insert a document here
48
  prompt = f"### Human: Please summarize the following article.\n\n{article}.\n###"
49
 
50
  inputs = tokenizer(header + prompt, return_tensors="pt")
51
- sample = model.generate(**inputs, max_new_tokens=128)
52
  print(tokenizer.decode(sample[0]))
53
  ```
54
 
 
48
  prompt = f"### Human: Please summarize the following article.\n\n{article}.\n###"
49
 
50
  inputs = tokenizer(header + prompt, return_tensors="pt")
51
+ sample = model.generate(**inputs, do_sample=True, max_new_tokens=2048, top_k=100, eos_token_id=50256)
52
  print(tokenizer.decode(sample[0]))
53
  ```
54