Skip to content

Commit 247d4e8

Browse files
author
Your Name
committed
update thresholds.json
1 parent fdaa0c2 commit 247d4e8

File tree

2 files changed

+633
-573
lines changed

2 files changed

+633
-573
lines changed

vllm/beam/filtering.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -22,13 +22,7 @@ def format_filter(filter_params_list):
2222

2323

2424
DEFAULT_CHAR_SERVER_FILTER = format_filter(
25-
[
26-
("annotations_porn", 0.1098),
27-
("annotations_racist", 0.2814),
28-
("annotations_disturbing", 0.1827),
29-
("annotations_harmful_promotes_selfharm", 0.0749),
30-
("annotations_harmful_promotes_terrorism", 0.1129),
31-
]
25+
[{"name": "annotations_porn", "threshold": 0.5156}, {"name": "annotations_racist", "threshold": 0.9763}, {"name": "annotations_disturbing", "threshold": 0.5472}, {"name": "annotations_harmful_promotes_selfharm", "threshold": 0.0657}]
3226
)
3327

3428
MAX_GENERATIONS = 10

0 commit comments

Comments
 (0)