From 457d3c58eb58d3450641e9695c55613ec62b23d2 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Tue, 11 Apr 2023 12:57:36 -0300 Subject: [PATCH] Update the API example --- api-example.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/api-example.py b/api-example.py index a3aaa0f0..23710047 100644 --- 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