oobabooga 2 роки тому
батько
коміт
457d3c58eb
1 змінених файлів з 2 додано та 0 видалено
  1. 2 0
      api-example.py

+ 2 - 0
api-example.py

@@ -35,6 +35,8 @@ params = {
     'length_penalty': 1,
     'length_penalty': 1,
     'early_stopping': False,
     'early_stopping': False,
     'seed': -1,
     'seed': -1,
+    'add_bos_token': True,
+    'custom_stopping_strings': [],
 }
 }
 
 
 # Input prompt
 # Input prompt