transformers
TF: XLA bad words logits processor and list of processors
#16974

Merged

TF: XLA bad words logits processor and list of processors #16974

gante merged 10 commits into huggingface:main from gante:xla_bad_words_ngrams

tmp commit

cfd8ccda

for future reference: ngram+xla like this gets a core dump if all_ngr…

ddc89115

while_loop doesn't seem to work either

74530f88

og test

40e3a413

gante changed the title ~~TF: XLA bad words and ngram logits processors~~ TF: XLA bad words logits processor and list of processors 4 years ago

revert ngram; add XLA test for the list of processors (and make corre…

accad57f

gante marked this pull request as ready for review 4 years ago

gante requested a review from

patrickvonplaten 4 years ago

gante requested a review from

Rocketknight1 4 years ago

typos

c14985a8

cur_len is argument everywhere

fbeae60c

typo

fc9972c6

patrickvonplaten commented on 2022-04-29

patrickvonplaten approved these changes on 2022-04-29

gante commented on 2022-04-29

Update src/transformers/generation_tf_logits_process.py

d4dcb3e1

gante commented on 2022-04-29

Update src/transformers/generation_tf_logits_process.py

4a69fba6

Rocketknight1 commented on 2022-04-29

Rocketknight1 approved these changes on 2022-04-29

gante merged fb0ae129 into main 4 years ago

gante deleted the xla_bad_words_ngrams branch 4 years ago

Reviewers

Rocketknight1

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone