• Abel's avatar
    Make T5 compatible with ONNX (#5518) · 69122657
    Abel authored
    
    
    * Default decoder inputs to encoder ones for T5 if neither are specified.
    
    * Fixing typo, now all tests are passing.
    
    * Changing einsum to operations supported by onnx
    
    * Adding a test to ensure T5 can be exported to onnx op>9
    
    * Modified test for onnx export to make it faster
    
    * Styling changes.
    
    * Styling changes.
    
    * Changing notation for matrix multiplication
    Co-authored-by: default avatarAbel Riboulot <tkai@protomail.com>
    69122657
test_modeling_t5.py 39.8 KB