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

Support ingesting PyTorch models #10

Open
pcolange opened this issue Dec 18, 2024 · 0 comments · May be fixed by #9
Open

Support ingesting PyTorch models #10

pcolange opened this issue Dec 18, 2024 · 0 comments · May be fixed by #9
Assignees
Labels
enhancement New feature or request

Comments

@pcolange
Copy link
Collaborator

This Issue is a feature request for the ability to load PyTorch models from a *.pth file into the GUI. Natively supporting PyTorch would require a new model summary GUI page, so we opt for this first pass to export PyTorch models as ONNX models so we can leverage all the current capabilities without requiring new PyTorch capabilities.

The story for this feature will include these steps:

  1. Enable *.pth files to be opened into Digest AI
  2. Open a new ingest window to enable users to set input dtypes and shapes and configure other export options
  3. Export the PyTorch model as an ONNX model
  4. Load exported ONNX model into Digest using the already supported ONNX model path
@pcolange pcolange added the enhancement New feature or request label Dec 18, 2024
@pcolange pcolange self-assigned this Dec 18, 2024
@pcolange pcolange linked a pull request Dec 18, 2024 that will close this issue
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant