Update the API example

2023-04-11 12:57:36 -03:00
parent 78bbc66fc4
commit 457d3c58eb
1 changed files with 2 additions and 0 deletions
--- a/api-example.py
+++ b/api-example.py
@@ -35,6 +35,8 @@ params = {
    'length_penalty': 1,
    'early_stopping': False,
    'seed': -1,
+    'add_bos_token': True,
+    'custom_stopping_strings': [],
 }

 # Input prompt