From 61641a4551cf06d9fafbfbc4162b8582ce3dc992 Mon Sep 17 00:00:00 2001 From: Alexander01998 Date: Wed, 12 Apr 2023 03:41:13 +0200 Subject: [PATCH] Add missing new parameters to API extension --- extensions/api/script.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/extensions/api/script.py b/extensions/api/script.py index bfce8d5b..44769027 100644 --- a/extensions/api/script.py +++ b/extensions/api/script.py @@ -59,6 +59,8 @@ class Handler(BaseHTTPRequestHandler): 'seed': int(body.get('seed', -1)), 'add_bos_token': int(body.get('add_bos_token', True)), 'custom_stopping_strings': body.get('custom_stopping_strings', []), + 'truncation_length': int(body.get('truncation_length', 2048)), + 'ban_eos_token': bool(body.get('ban_eos_token', False)), } generator = generate_reply(