oobabooga changed api

2 years ago · 7cf7724f29
1 changed files with 1 additions and 1 deletions
--- 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,