• Patrick von Platen's avatar
    TF generate refactor - Greedy Search (#15562) · 2e12b907
    Patrick von Platen authored
    
    
    * TF generate start refactor
    
    * Add tf tests for sample generate
    
    * re-organize
    
    * boom boom
    
    * Apply suggestions from code review
    
    * re-add
    
    * add all code
    
    * make random greedy pass
    
    * make encoder-decoder random work
    
    * further improvements
    
    * delete bogus file
    
    * make gpt2 and t5 tests work
    
    * finish logits tests
    
    * correct logits processors
    
    * correct past / encoder_outputs drama
    
    * refactor some methods
    
    * another fix
    
    * refactor shape_list
    
    * fix more shape list
    
    * import shape
    _list
    
    * finish docs
    
    * fix imports
    
    * make style
    
    * correct tf utils
    
    * Fix TFRag as well
    
    * Apply Lysandre's and Sylvais suggestions
    
    * Update tests/test_generation_tf_logits_process.py
    Co-authored-by: default avatarMatt <Rocketknight1@users.noreply.github.com>
    
    * Update src/transformers/tf_utils.py
    Co-authored-by: default avatarMatt <Rocketknight1@users.noreply.github.com>
    
    * remove cpu according to gante
    
    * correct logit processor
    Co-authored-by: default avatarMatt <Rocketknight1@users.noreply.github.com>
    2e12b907
test_modeling_tf_common.py 65.3 KB