Browse Source

oobabooga changed api

master
Hendrik Langer 2 years ago
parent
commit
7cf7724f29
  1. 2
      matrix_pygmalion_bot/ai/runpod.py

2
matrix_pygmalion_bot/ai/runpod.py

@ -66,7 +66,7 @@ async def generate_sync(
'early_stopping': False, 'early_stopping': False,
'seed': -1, 'seed': -1,
'add_bos_token': True, 'add_bos_token': True,
'custom_stopping_strings': [f"\n{bot.user_name}:"], 'stopping_strings': [f"\n{bot.user_name}:"],
'truncation_length': 2048, 'truncation_length': 2048,
'ban_eos_token': False, 'ban_eos_token': False,
'skip_special_tokens': True, 'skip_special_tokens': True,

Loading…
Cancel
Save