"sgl-router/src/vscode:/vscode.git/clone" did not exist on "00eb5eb7213c7b75984ef44eae80bda2228d4cfd"
  • amyeroberts's avatar
    Update serving code to enable `saved_model=True` (#18153) · 8e838466
    amyeroberts authored
    
    
    * Add serving_output and serving methods to some vision models
    
    * Add serving outputs for DeiT
    
    * Don't convert hidden states - differing shapes
    
    * Make saveable
    
    * Fix up
    
    * Make swin saveable
    
    * Add in tests
    
    * Fix funnel tests (can't convert to tensor)
    
    * Fix numpy call
    
    * Tidy up a bit
    
    * Add in hidden states - resnet
    
    * Remove numpy
    
    * Fix failing tests - tensor shape and skipping tests
    
    * Remove duplicated function
    
    * PR comments - formatting and var names
    
    * PR comments
    Add suggestions made by Joao Gante:
    * Use tf.shape instead of shape_list
    * Use @tooslow decorator on tests
    * Simplify some of the logic
    
    * PR comments
    Address Yih-Dar Sheih comments - making tensor names consistent and make types float
    
    * Types consistent with docs; disable test on swin (slow)
    
    * CI trigger
    
    * Change input_features to float32
    
    * Add serving_output for segformer
    
    * Fixup
    Co-authored-by: default avatarAmy Roberts <amyeroberts@users.noreply.github.com>
    8e838466
test_modeling_tf_t5.py 55.6 KB