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

Onnx - Load external data from memory #1355

Open
kckeiks opened this issue Mar 13, 2024 · 2 comments
Open

Onnx - Load external data from memory #1355

kckeiks opened this issue Mar 13, 2024 · 2 comments

Comments

@kckeiks
Copy link

kckeiks commented Mar 13, 2024

Hello,
Awesome project!
It would be great to have this feature. In our project, a process loads models fetched from the network but we're currently restricted to only supporting models <2GB because storing/loading models from disk is not an option for us.
Cheers!

@kckeiks kckeiks changed the title Load external data from memory Onnx - Load external data from memory Mar 13, 2024
@kali
Copy link
Collaborator

kali commented Mar 22, 2024

Hey, thanks for your interest!

Could you have a look at #1358 ? I think by replacing the data_provider on the Onnx structure by your own implementation, you would be able to redirect the data access to wherever you want. Please tell me what you think.

@kckeiks
Copy link
Author

kckeiks commented Mar 24, 2024

Hey, thanks for your interest!

Could you have a look at #1358 ? I think by replacing the data_provider on the Onnx structure by your own implementation, you would be able to redirect the data access to wherever you want. Please tell me what you think.

Thanks for taking the time, it looks good! I think it will work for our case since we have the model and external data in memory so we could just implement ModelDataResolver. Looking forward to trying it!

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

No branches or pull requests

2 participants