• Jeremy Fisher's avatar
    Fix and improve CTRL doctests (#16573) · 0235bc57
    Jeremy Fisher authored
    
    
    * Improve CTRL doctests
    
    * Fix `CTRLForSequenceClassification` flakiness with inconsistent losses
    
    * Remove unused
    
    * Fixup
    
    * Add CTRL to documentation_tests.txt
    
    * Fix control code not being first
    
    * Add output assertions
    
    * Change from sshleifer/tiny-ctrl -> ctrl
    
    * Run `make fixup`
    
    * apply `list` to output logits shape for clarity
    
    * Reduce output loss precision to make assertion more robust
    
    * Add assertion of control code being first
    
    * Fix docstyle
    
    * upper case sentence following control code
    
    * Weird bug fixes
    
    * Add a better generation example
    Co-authored-by: default avatarYih-Dar <2521628+ydshieh@users.noreply.github.com>
    0235bc57
documentation_tests.txt 2.75 KB