Skip to content

Commit

Permalink
precommit formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
agentd00nut committed Dec 3, 2023
1 parent 37f2250 commit 26477e0
Showing 1 changed file with 11 additions and 6 deletions.
17 changes: 11 additions & 6 deletions novelai_api/_high_level.py
Original file line number Diff line number Diff line change
Expand Up @@ -298,12 +298,17 @@ async def _generate(
global_settings.rep_pen_whitelist = repetition_penalty_default_whitelist

params = {
"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
))
"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)
Expand Down

0 comments on commit 26477e0

Please sign in to comment.