• Patrick Deutschmann's avatar
    Add ONNX support for Longformer (#17176) · 3223d493
    Patrick Deutschmann authored
    * Implement ONNX support for Longformer
    
    Fix repo consistency check complaints
    
    Fix value mismatches
    
    Add pooler output for default model
    
    Increase validation atol to accommodate multiple-choice error
    
    Fix copies
    
    Fix chunking for longer sequence lengths
    
    Add future comment
    
    * Fix issue in mask_invalid_locations
    
    * Remove torch imports in configuration_longformer
    
    * Change config access to fix LED
    
    * Push opset version to support tril
    
    * Work in review comments (mostly style)
    
    * Add Longformer to ONNX tests
    3223d493
serialization.mdx 26.1 KB