From 7cf7724f293e7eca83914ccd00fb1a4506f5268b Mon Sep 17 00:00:00 2001 From: Hendrik Langer Date: Thu, 20 Apr 2023 18:04:44 +0200 Subject: [PATCH] oobabooga changed api --- matrix_pygmalion_bot/ai/runpod.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/matrix_pygmalion_bot/ai/runpod.py b/matrix_pygmalion_bot/ai/runpod.py index b34931a..08f5d25 100644 --- a/matrix_pygmalion_bot/ai/runpod.py +++ b/matrix_pygmalion_bot/ai/runpod.py @@ -66,7 +66,7 @@ async def generate_sync( 'early_stopping': False, 'seed': -1, 'add_bos_token': True, - 'custom_stopping_strings': [f"\n{bot.user_name}:"], + 'stopping_strings': [f"\n{bot.user_name}:"], 'truncation_length': 2048, 'ban_eos_token': False, 'skip_special_tokens': True,