Shaltiel commited on
Commit
cbc604b
1 Parent(s): 039894b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -0
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