erythropygia commited on
Commit
5c03886
·
verified ·
1 Parent(s): 6cc55ed

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -66,7 +66,7 @@ inputs = tokenizer(
66
 
67
  from transformers import TextStreamer
68
  text_streamer = TextStreamer(tokenizer)
69
- _ = model.generate(**inputs, streamer = text_streamer, max_new_tokens = 256)
70
  ```
71
 
72
  ## Performance
 
66
 
67
  from transformers import TextStreamer
68
  text_streamer = TextStreamer(tokenizer)
69
+ _ = model.generate(**inputs, streamer = text_streamer, max_new_tokens = 512, do_sample=True, temperature=0.75, top_k=50, top_p=0.9, repetition_penalty=1.1)
70
  ```
71
 
72
  ## Performance