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

Does wasmtime support AI model inference? I want to perform resnet50.pth inference in wasmtime, but there is no relevant tutorial content in the document. #8361

Open
lebron8dong opened this issue Apr 13, 2024 · 2 comments

Comments

@lebron8dong
Copy link

Are there any related examples demo?
Thank you so much.

@bjorn3
Copy link
Contributor

bjorn3 commented Apr 13, 2024

There is the wasi-nn specification implemented by wasmtime. https://github.com/bytecodealliance/wasmtime/tree/main/crates/wasi-nn/examples has a couple of examples.

@rockwotj
Copy link
Contributor

It's also possible to compile many models to Wasm and run inference. For example, https://github.com/sonos/tract/ compiles to WebAssembly & WASI and can run any ONNX compatible model.

I have an example of running a BERT model here: https://github.com/rockwotj/ai-qa-wasi/tree/repl

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

3 participants