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.

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

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

Conversation

srmsoumya
Copy link
Collaborator

@srmsoumya srmsoumya commented Jul 18, 2024

  • Add code to take the clay encoder & compile it to torch exported program format
  • Add code to compile torch encoder into onnx format
  • Notebook to show how to use the compiled models

Note: onnx gpu export is not implemented, will fix that in a new PR.

src/utils.py Outdated Show resolved Hide resolved
@srmsoumya srmsoumya marked this pull request as ready for review July 25, 2024 12:53
@srmsoumya srmsoumya requested a review from yellowcap July 25, 2024 12:55
@srmsoumya
Copy link
Collaborator Author

@rbavery This PR adds compiled version of Clay v1 Encoder in ONNX & torch ExportedProgram format.

@lauracchen
Copy link
Member

@srmsoumya We are still missing the onnx GPU export right? I'm only able to run onnx for some reason, so it'd be great to have access to the GPU version too (and I'm guessing maybe others might have the same experience)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants