Skip to content

Update to latest RustNN API#14

Merged
mtavenrath merged 1 commit intomainfrom
flux2_prepare
Apr 28, 2026
Merged

Update to latest RustNN API#14
mtavenrath merged 1 commit intomainfrom
flux2_prepare

Conversation

@mtavenrath
Copy link
Copy Markdown
Contributor

  • Move from String to MLOperandDataType
  • Support new split protobuf + weights API for ONNX
  • Use webnn-graph dataloader

@mtavenrath mtavenrath marked this pull request as draft April 24, 2026 13:54
Copy link
Copy Markdown
Contributor

@tarekziade tarekziade left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

look pretty mechanical +1 once cleaned up

Comment thread Cargo.toml Outdated
serde_json = "1.0"
webnn-graph = { git = "https://github.com/rustnn/webnn-graph", branch = "main" }
safetensors = "0.7"
#webnn-graph = { path = "../webnn-graph" }
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

to be removed before landing

Comment thread Cargo.toml Outdated
pyo3 = { version = "0.28", features = ["extension-module"] }
rustnn = { git = "https://github.com/rustnn/rustnn", branch = "main" }
rustnn = { git = "https://github.com/rustnn/rustnn", branch = "flux2_prepare" }
#rustnn = { path = "../rustnn" }
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

tpo be remove

@mtavenrath mtavenrath marked this pull request as ready for review April 28, 2026 12:45
@mtavenrath mtavenrath merged commit cf52e19 into main Apr 28, 2026
29 of 44 checks passed
@mtavenrath mtavenrath deleted the flux2_prepare branch April 28, 2026 14:23
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.

2 participants