From 2ad60d94833f8366271540d6d0393ccbb2b8b4f5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fr=C3=A9d=C3=A9ric=20Bolvin?= Date: Sat, 25 Nov 2023 22:13:30 +0100 Subject: [PATCH] flatten repetition_penalty_whitelist --- novelai_api/_high_level.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/novelai_api/_high_level.py b/novelai_api/_high_level.py index a7c91e8..32a2b09 100644 --- a/novelai_api/_high_level.py +++ b/novelai_api/_high_level.py @@ -298,10 +298,12 @@ async def _generate( global_settings.rep_pen_whitelist = repetition_penalty_default_whitelist params = { - "repetition_penalty_whitelist": [ - *global_params.pop("repetition_penalty_whitelist", []), - *preset_params.pop("repetition_penalty_whitelist", []), - ] + "repetition_penalty_whitelist": list(set( + item for sublist in [ + global_params.pop("repetition_penalty_whitelist", []), + preset_params.pop("repetition_penalty_whitelist", []), + ] for inner_list in sublist for item in inner_list + )) } params.update(preset_params)