File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed
Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -272,6 +272,7 @@ class ChatCompletionRequest(OpenAIBaseModel):
272272 truncate_prompt_tokens : Optional [Annotated [int , Field (ge = 1 )]] = None
273273 prompt_logprobs : Optional [int ] = None
274274 allowed_token_ids : Optional [list [int ]] = None
275+ bad_words : list [str ] = Field (default_factory = list )
275276 # --8<-- [end:chat-completion-sampling-params]
276277
277278 # --8<-- [start:chat-completion-extra-params]
@@ -550,6 +551,7 @@ def to_sampling_params(
550551 else RequestOutputKind .FINAL_ONLY ,
551552 guided_decoding = guided_decoding ,
552553 logit_bias = self .logit_bias ,
554+ bad_words = self .bad_words ,
553555 allowed_token_ids = self .allowed_token_ids ,
554556 extra_args = ({"kv_transfer_params" : self .kv_transfer_params }
555557 if self .kv_transfer_params else None ))
You can’t perform that action at this time.
0 commit comments