riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix description and formatting errors in code docs (#38074)

Browse Source 
* Update stopping_criteria.py

Fix description and formatting errors.

* Update stopping_criteria.py

Align formatting with existing files for consistency.
This commit is contained in:
bilibili12433014 2025-05-14 01:17:15 +08:00 committed by GitHub
parent b499a14b17
commit b311a3f506
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/transformers/generation/stopping_criteria.py
View File

@ -37,8 +37,9 @@ STOPPING_CRITERIA_INPUTS_DOCSTRING = r"""
Additional stopping criteria specific kwargs.
Return:
`torch.BoolTensor`. (`torch.BoolTensor` of shape `(batch_size, 1)`), where `True` indicates we stop generation
for a particular row, `True` indicates we should continue.
`torch.BoolTensor`. (`torch.BoolTensor` of shape `(batch_size, 1)`):
`True` indicates we stop generation for a particular row.
`False` indicates we should continue.
"""