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

Add scripts to export the encoder of Clay #303

Open
wants to merge 12 commits into
base: main
Choose a base branch
from
Open

Add scripts to export the encoder of Clay #303

wants to merge 12 commits into from

Commits on Jul 25, 2024

  1. Compile clay model encoder

    @srmsoumya
    srmsoumya committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    48748f1 View commit details
    Browse the repository at this point in the history

  2. Add benchmark & test files for the compiled clay encoder

    @srmsoumya
    srmsoumya committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    97eb19a View commit details
    Browse the repository at this point in the history

  3. Revert changes to Encoder, don't change the API

    @srmsoumya
    srmsoumya committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    eba1867 View commit details
    Browse the repository at this point in the history

  4. Add embedder to load clay encoder & save in onnx/ep format

    @srmsoumya
    srmsoumya committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    73171dd View commit details
    Browse the repository at this point in the history

  5. Remove files from src, fix utils to run everything on same device

    @srmsoumya
    srmsoumya committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    1f2fcc9 View commit details
    Browse the repository at this point in the history

  6. Bump torch==2.3.1 & torchvision==0.18.1, add onnx & onnxsxript as dep… 

    …endency
    @srmsoumya
    srmsoumya committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    0a3ce9d View commit details
    Browse the repository at this point in the history

  7. Release few contraints on env

    @srmsoumya
    srmsoumya committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    37503fe View commit details
    Browse the repository at this point in the history

  8. Add notebook to show how to embed using compiled embedders

    @srmsoumya
    srmsoumya committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    db8a3f2 View commit details
    Browse the repository at this point in the history

  9. [pre-commit.ci] auto fixes from pre-commit.com hooks 

    for more information, see https://pre-commit.ci
    @pre-commit-ci
    pre-commit-ci[bot] committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    c0552bd View commit details
    Browse the repository at this point in the history

  10. Clear outputs from the notebook

    @srmsoumya
    srmsoumya committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    1e97506 View commit details
    Browse the repository at this point in the history

  11. Merge branch 'compile' of https://github.com/Clay-foundation/model in… 

    …to compile
    @srmsoumya
    srmsoumya committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    0c37c58 View commit details
    Browse the repository at this point in the history

  12. Add torchdata as a pip dependency

    @srmsoumya
    srmsoumya committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    1803954 View commit details
    Browse the repository at this point in the history