Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Patrick von Platen's avatar
    [Generate] Facilitate PyTorch generate using `ModelOutputs` (#6735) · afc4ece4
    Patrick von Platen authored 
    * fix generate for GPT2 Double Head

* fix gpt2 double head model

* fix  bart / t5

* also add for no beam search

* fix no beam search

* fix encoder decoder

* simplify t5

* simplify t5

* fix t5 tests

* fix BART

* fix transfo-xl

* fix conflict

* integrating sylvains and sams comments

* fix tf past_decoder_key_values

* fix enc dec test
    afc4ece4
test_modeling_openai.py 7.71 KB