Skip to content

Commit

Permalink
remove print
Browse files Browse the repository at this point in the history
  • Loading branch information
bastiscode committed Feb 12, 2024
1 parent cd03b09 commit d0dc4dc
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion python/text_utils/inference/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -260,7 +260,6 @@ def search(
)[:, :max_decoder_length] # type: ignore
# always add a padding mask, indicating which tokens are padding
# and the lengths of the sequence to the additional arguments
print(list(decoder_kwargs))
assert "padding_mask" not in decoder_kwargs and "lengths" not in decoder_kwargs, \
"padding_mask and lengths are added automatically, do not provide them yourself"
decoder_kwargs["padding_mask"] = decoder_token_ids == pad_token_id
Expand Down

0 comments on commit d0dc4dc

Please sign in to comment.