"benchmark/kernels/vscode:/vscode.git/clone" did not exist on "de2dd73831bdd22b5f2946909e6827ece4712ba8"
  • lewtun's avatar
    Add ONNX export for ViT (#15658) · 50dd314d
    lewtun authored
    
    
    * Add ONNX support for ViT
    
    * Refactor to use generic preprocessor
    
    * Add vision dep to tests
    
    * Extend ONNX slow tests to ViT
    
    * Add dummy image generator
    
    * Use model_type to determine modality
    
    * Add deprecation warnings for tokenizer argument
    
    * Add warning when overwriting the preprocessor
    
    * Add optional args to docstrings
    
    * Add minimum PyTorch version to OnnxConfig
    
    * Refactor OnnxConfig class variables from CONSTANT_NAME to snake_case
    
    * Add reasonable value for default atol
    Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
    50dd314d
serialization.mdx 25.6 KB