We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7867558 commit c188498Copy full SHA for c188498
src/transformers/generation/utils.py
@@ -953,8 +953,7 @@ def _get_stopping_criteria(
953
if generation_config._eos_token_tensor is not None:
954
criteria.append(EosTokenCriteria(eos_token_id=generation_config._eos_token_tensor))
955
if (
956
- generation_config.is_assistant is not None
957
- and generation_config.is_assistant
+ generation_config.is_assistant
958
and generation_config.assistant_confidence_threshold is not None
959
and generation_config.assistant_confidence_threshold > 0
960
):
0 commit comments