Rename default model name to model.onnx.

It aligns with openvino backend. prtest:full
bytecodealliance · Feb 1, 2024 · 9732234 · 9732234
1 parent cf65b2e
commit 9732234
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/crates/wasi-nn/src/backend/winml.rs b/crates/wasi-nn/src/backend/winml.rs
@@ -56,7 +56,7 @@ impl BackendFromDir for WinMLBackend {
         path: &Path,
         target: ExecutionTarget,
     ) -> Result<Graph, BackendError> {
-        let model = read(&path.join("mobilenet.onnx"))?;
+        let model = read(&path.join("model.onnx"))?;
         self.load(&[&model], target)
     }
 }

diff --git a/crates/wasi-nn/src/testing.rs b/crates/wasi-nn/src/testing.rs
@@ -97,7 +97,7 @@ fn check_winml_artifacts_are_available() -> Result<()> {
         fs::create_dir(&artifacts_dir)?;
     }
     const MODEL_URL: &str="https://github.com/onnx/models/raw/5faef4c33eba0395177850e1e31c4a6a9e634c82/vision/classification/mobilenet/model/mobilenetv2-12.onnx";
-    for (from, to) in [(MODEL_URL, "mobilenet.onnx")] {
+    for (from, to) in [(MODEL_URL, "model.onnx")] {
         let local_path = artifacts_dir.join(to);
         if !local_path.is_file() {
             download(&from, &local_path).with_context(|| "unable to retrieve test artifact")?;