Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[CLIP]: Sparseml onnx export pathways for open_clip models #1626

Merged
merged 11 commits into from
Jun 23, 2023
Merged

[CLIP]: Sparseml onnx export pathways for open_clip models #1626

merged 11 commits into from
Jun 23, 2023

Commits on Jun 21, 2023

  1. Add example script on how to export open_clip models using sparseml o… 

    …nnx export; add additional models to encapsulate operations needed during inference time
    @dsikka
    dsikka committed Jun 21, 2023
    Configuration menu
    Copy the full SHA
    f9485ae View commit details
    Browse the repository at this point in the history

  2. update setup.py to add clip dependency, add copyright

    @dsikka
    dsikka committed Jun 21, 2023
    Configuration menu
    Copy the full SHA
    a01f8f3 View commit details
    Browse the repository at this point in the history

  3. format fixes

    @dsikka
    dsikka committed Jun 21, 2023
    Configuration menu
    Copy the full SHA
    3f9aaec View commit details
    Browse the repository at this point in the history

  4. add docstring and example cli command

    @dsikka
    dsikka committed Jun 21, 2023
    Configuration menu
    Copy the full SHA
    226ca14 View commit details
    Browse the repository at this point in the history

  5. Add decoder export; update docstring

    @dsikka
    dsikka committed Jun 21, 2023
    Configuration menu
    Copy the full SHA
    0ceec8f View commit details
    Browse the repository at this point in the history

  6. format fix

    @dsikka
    dsikka committed Jun 21, 2023
    Configuration menu
    Copy the full SHA
    28bd2e5 View commit details
    Browse the repository at this point in the history

  7. remove tokenizer and transformations not used in the forward pass

    @dsikka
    dsikka committed Jun 21, 2023
    Configuration menu
    Copy the full SHA
    88cacfa View commit details
    Browse the repository at this point in the history

  8. update readme to bumpup pytorch version; add torch nightly caveat for… 

    … clip
    @dsikka
    dsikka committed Jun 21, 2023
    Configuration menu
    Copy the full SHA
    3cdacb1 View commit details
    Browse the repository at this point in the history

Commits on Jun 22, 2023

  1. Update README, use env variable for max torch, add typing

    @dsikka
    dsikka committed Jun 22, 2023
    Configuration menu
    Copy the full SHA
    a2907bd View commit details
    Browse the repository at this point in the history

Commits on Jun 23, 2023

  1. Merge branch 'main' into clip_export_example

    @dsikka
    dsikka authored Jun 23, 2023
    Configuration menu
    Copy the full SHA
    68f756a View commit details
    Browse the repository at this point in the history

  2. throw an error if opset < 14 is used

    @dsikka
    dsikka committed Jun 23, 2023
    Configuration menu
    Copy the full SHA
    980b6bc View commit details
    Browse the repository at this point in the history