Update README.md
Browse files
README.md
CHANGED
@@ -55,6 +55,10 @@ with torch.inference_mode():
|
|
55 |
print(tokenizer.batch_decode(model.generate(**kwargs), skip_special_tokens=True))
|
56 |
```
|
57 |
|
|
|
|
|
|
|
|
|
58 |
|
59 |
## Citation
|
60 |
|
|
|
55 |
print(tokenizer.batch_decode(model.generate(**kwargs), skip_special_tokens=True))
|
56 |
```
|
57 |
|
58 |
+
There are many different parameters you can input into `kwargs` for different results (greedy, beamsearch, different samplign configurations, longer/shorter respones, etc.).
|
59 |
+
|
60 |
+
You can view the full list of parameters you can pass to the `generate` function [here](https://huggingface.co/docs/transformers/v4.33.0/en/main_classes/text_generation#transformers.GenerationMixin.generate).
|
61 |
+
|
62 |
|
63 |
## Citation
|
64 |
|