• Patrick von Platen's avatar
    Refactoring the generate() function (#6949) · a1bbcf3f
    Patrick von Platen authored
    * first draft
    
    * show design proposition for new generate method
    
    * up
    
    * make better readable
    
    * make first version
    
    * gpt2 tests pass
    
    * make beam search for gpt2 work
    
    * add first encoder-decoder code
    
    * delete typo
    
    * make t5 work
    
    * save indermediate
    
    * make bart work with beam search
    
    * finish beam search bart / t5
    
    * add default kwargs
    
    * make more tests pass
    
    * fix no bad words sampler
    
    * some fixes and tests for all distribution processors
    
    * fix test
    
    * fix rag slow tests
    
    * merge to master
    
    * add nograd to generate
    
    * make all slow tests pass
    
    * speed up generate
    
    * fix edge case bug
    
    * small fix
    
    * correct typo
    
    * add type hints and docstrings
    
    * fix typos in tests
    
    * add beam search tests
    
    * add tests for beam scorer
    
    * fix test rag
    
    * finish beam search tests
    
    * move generation tests in seperate file
    
    * fix generation tests
    
    * more tests
    
    * add aggressive generation tests
    
    * fix tests
    
    * add gpt2 sample test
    
    * add more docstring
    
    * add more docs
    
    * finish doc strings
    
    * apply some more of sylvains and sams comments
    
    * fix some typos
    
    * make fix copies
    
    * apply lysandres and sylvains comments
    
    * final corrections on examples
    
    * small fix for reformer
    a1bbcf3f
test_modeling_t5.py 45.5 KB