Skip to content

Commit c188498

Browse files
jmamougante
andcommitted
Update src/transformers/generation/utils.py
Co-authored-by: Joao Gante <[email protected]>
1 parent 7867558 commit c188498

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/transformers/generation/utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -953,8 +953,7 @@ def _get_stopping_criteria(
953953
if generation_config._eos_token_tensor is not None:
954954
criteria.append(EosTokenCriteria(eos_token_id=generation_config._eos_token_tensor))
955955
if (
956-
generation_config.is_assistant is not None
957-
and generation_config.is_assistant
956+
generation_config.is_assistant
958957
and generation_config.assistant_confidence_threshold is not None
959958
and generation_config.assistant_confidence_threshold > 0
960959
):

0 commit comments

Comments
 (0)