• Kevin Bondzio's avatar
    Add soft length regulation for sequence generation (#15245) · 9442b3ce
    Kevin Bondzio authored
    
    
    * add possibility to softly regulate length when using sampling method in model.generate() function
    
    * fix test config, fix formatting
    
    * fix rag integration, fix docstyling
    
    * fix wrong docstring
    
    * change param to tuple, add test
    
    * fix old param in rag_model, remove unused import
    
    * change test according to new param
    
    * fix formatting
    
    * fix test case
    
    * fix doc style
    
    * move start_length calculation to Logitprocessor
    
    * add possibility to softly regulate length when using sampling method in model.generate() function
    
    * fix rag integration, fix docstyling
    
    * fix test config, fix formatting
    
    * change param to tuple, add test
    
    * fix old param in rag_model, remove unused import
    
    * add possibility to softly regulate length when using sampling method in model.generate() function
    
    * change param to tuple, add test
    
    * fix old param in rag_model, remove unused import
    
    * remove unused import
    
    * fix small errors
    
    * fix test
    
    * add possibility to softly regulate length when using sampling method in model.generate() function
    
    * fix test config, fix formatting
    
    * fix rag integration, fix docstyling
    
    * change param to tuple, add test
    
    * fix old param in rag_model, remove unused import
    
    * change test according to new param
    
    * fix test case
    
    * move start_length calculation to Logitprocessor
    
    * add possibility to softly regulate length when using sampling method in model.generate() function
    
    * fix rag integration, fix docstyling
    
    * fix test config, fix formatting
    
    * change param to tuple, add test
    
    * fix old param in rag_model, remove unused import
    
    * add possibility to softly regulate length when using sampling method in model.generate() function
    
    * fix test config, fix formatting
    
    * fix rag integration, fix docstyling
    
    * add possibility to softly regulate length when using sampling method in model.generate() function
    
    * fix rag integration, fix docstyling
    
    * change param to tuple, add test
    
    * fix old param in rag_model, remove unused import
    
    * fix small errors
    
    * Update src/transformers/generation_utils.py
    Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
    
    * Update src/transformers/generation_utils.py
    
    * Update src/transformers/generation_utils.py
    
    * fix docstring, add type ind model rag
    
    * fix docstrings
    
    * introduce seq_length variable for cleaner code
    
    * fix black formatting
    
    * add input_ids_seq_length to modeling_rag
    
    * add input_ids_seq_length to test
    
    * retrigger checks
    
    * retrigger checks
    Co-authored-by: default avatarKevin Bondzio <kev@AIM-LAP-02.local>
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
    Co-authored-by: default avatarKevin Bondzio <kev@AIM-LAP-02.fritz.box>
    9442b3ce
test_configuration_common.py 14.3 KB