"INSTALL/vscode:/vscode.git/clone" did not exist on "b7e878c466712712f47956d97dd7e33d0a6e0cef"
  • Chan Woo Kim's avatar
    Constrained Beam Search [without disjunctive decoding] (#15416) · 2b5603f6
    Chan Woo Kim authored
    
    
    * added classes to get started with constrained beam search
    
    * in progress, think i can directly force tokens now but not yet with the round robin
    
    * think now i have total control, now need to code the bank selection
    
    * technically works as desired, need to optimize and fix design choices leading to undersirable outputs
    
    * complete PR #1 without disjunctive decoding
    
    * removed incorrect tests
    
    * Delete k.txt
    
    * Delete test.py
    
    * Delete test.sh
    
    * revert changes to test scripts
    
    * genutils
    
    * full implementation with testing, no disjunctive yet
    
    * shifted docs
    
    * passing all tests realistically ran locally
    
    * removing accidentally included print statements
    
    * fixed source of error in initial PR test
    
    * fixing the get_device() vs device trap
    
    * fixed documentation docstrings about constrained_beam_search
    
    * fixed tests having failing for Speech2TextModel's floating point inputs
    
    * fix cuda long tensor
    
    * added examples and testing for them and founx & fixed a bug in beam_search and constrained_beam_search
    
    * deleted accidentally added test halting code with assert False
    
    * code reformat
    
    * Update tests/test_generation_utils.py
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    
    * Update tests/test_generation_utils.py
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    
    * Update tests/test_generation_utils.py
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    
    * Update tests/test_generation_utils.py
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    
    * Update tests/test_generation_utils.py
    
    * fixing based on comments on PR
    
    * took out the testing code that should but work fails without the beam search moditification ; style changes
    
    * fixing comments issues
    
    * docstrings for ConstraintListState
    
    * typo in PhrsalConstraint docstring
    
    * docstrings improvements
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    2b5603f6
test_generation_beam_search.py 22.7 KB