Update README.md
Browse files
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 =
|
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
|