"tests/pipelines/chroma/test_pipeline_chroma_img2img.py" did not exist on "0e460675e2c886ac4e9e8b09704be2a4d8aae15d"
  • 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_blenderbot.py 13.1 KB