transformers
Fix contrastive search to correctly handle input with padding
#33507
Merged
Merged
Fix contrastive search to correctly handle input with padding #33507
gante
merged
6 commits into
huggingface:main
from
fix-contrastive-search-inconsistency
ducviet00
commented
on 2024-09-17
fix: handle padding in contrastive search for decoder-only models
055b692b
fix: handle padding in contrastive search for encoder-decoder models
daa26963
tests: move padding contrastive test to test_util, add t5 test
a10ec988
fix: handle if model_kwargs["decoder_attention_mask"] is None
6a86c0d5
refactor: improve padding input contrastive search generation tests
b8a08dd7
chore: _ranking_fast to use LongTensor for cosine_matrix_mask
21d0207d
ducviet00
changed the title
ducviet00
changed the title
gante
approved these changes
on 2024-09-19
LysandreJik
approved these changes
on 2024-09-20
Login to write a write a comment.
Login via GitHub