mlboydaisuke commited on
Commit
5f6a23e
·
verified ·
1 Parent(s): f482fc5

Upload Vision-RWKV-S-OpenVINO: ONNX, OpenVINO IR, Intel NPU blob, card and manifest

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ npu/vrwkv_s_fp16.blob filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: litert-community/Vision-RWKV-S-LiteRT
4
+ base_model_relation: quantized
5
+ library_name: openvino
6
+ pipeline_tag: image-classification
7
+ tags:
8
+ - openvino
9
+ - onnx
10
+ - intel-npu
11
+ ---
12
+
13
+ # Vision-RWKV-S-OpenVINO
14
+
15
+ Vision-RWKV-S for Intel NPUs, GPUs and CPUs: vrwkv_s_fp16 from [litert-community/Vision-RWKV-S-LiteRT](https://huggingface.co/litert-community/Vision-RWKV-S-LiteRT/tree/184992fed3c7bb2761220287bcd0a496959e4e39) as ONNX (tf2onnx 1.17.0, opset 17), OpenVINO IR with fp32 weights, and a precompiled Intel NPU blob. On the NPU of a Core Ultra 9 386H the blob runs in 234.02 ms per run, numpy input to numpy output.
16
+
17
+ Scripts, raw records, and the same table for all 52 graphs of LiteRT-Models: https://github.com/john-rocky/windows-zoo
18
+
19
+ ## Files
20
+
21
+ | file | bytes | sha256 |
22
+ |---|---|---|
23
+ | `vrwkv_s_fp16.onnx` | 95,635,055 | `12184150736b7ec227a0a98ca6fa70ab69c8894ecb9cba800ce2537c6d75aaed` |
24
+ | `openvino/vrwkv_s_fp16.xml` | 1,546,499 | `f34f101344f8e83faddd5c81987249a028057f284edd41057308e11907390d5e` |
25
+ | `openvino/vrwkv_s_fp16.bin` | 95,336,452 | `54b97b3387a49b24689b5e78ad3d6004d78b742d327c30805b76a802ffa95782` |
26
+ | `npu/vrwkv_s_fp16.blob` | 74,580,104 | `bed2c0e5fe4cc71621824d94642135ec77a1b0ca3bd384f861294a95bfaf76fe` |
27
+
28
+ The NPU blob was compiled with OpenVINO 2026.4.1, Intel NPU driver 32.0.100.4841 (OpenVINO reports NPU_DRIVER_VERSION 1004841, NPU_COMPILER_VERSION 524292). If `import_model` refuses a blob on another driver or OpenVINO version, compile the IR instead (the commented line in the snippet).
29
+
30
+ ## How it was measured
31
+
32
+ MSI Stealth 16 AI+ (Core Ultra 9 386H, RTX 5080 Laptop GPU at a 95 W power limit, Windows 11 build 26200.9711, on AC power), 2026-11-06 to 2026-11-07. On the laptop, each row is one process: load, one run, 5 warmup runs, then the median of 30 timed runs (2 warmup + 10 timed runs when the first run takes more than 3 s), numpy input to numpy output, every runtime at its default settings, one process at a time, each started after the CPU, the RTX and the Intel GPU had gone idle (the thresholds are in METHOD.md of the GitHub repo). Output check: each output against the CPU output in fp32 on the same input (ONNX Runtime's CPU EP); PASS = cosine ≥ 0.99, no NaN or Inf, and top-1 agreement ≥ 0.99 for class scores. The Snapdragon rows are Qualcomm AI Hub profile jobs of the same ONNX files (their own timer).
33
+
34
+ ## Speed and output check
35
+
36
+ | processor | runtime | file | ms | load s | output check |
37
+ |---|---|---|---|---|---|
38
+ | Intel AI Boost NPU | OpenVINO 2026.4.1 | `npu/` blob (`import_model`) | 234.02 | 0.05 | PASS (cos ≥ 0.9999) |
39
+ | Intel AI Boost NPU | ONNX Runtime 1.31.1 + OpenVINO EP 1.8.1 | `.onnx`, compiled when the session loads | 282.56 | 67.01 | PASS (cos ≥ 0.9999) |
40
+ | Intel Graphics (iGPU) | ONNX Runtime 1.31.1 + OpenVINO EP 1.8.1 | `.onnx` | 51.47 | 3.44 | PASS (cos ≥ 0.9999) |
41
+ | RTX 5080 Laptop (95 W limit) | ONNX Runtime 1.31.1, CUDA EP | fp16 copy of the `.onnx` (not in this repo) | 15.01 | 1.27 | PASS (cos ≥ 0.9998) |
42
+ | RTX 5080 Laptop (95 W limit) | ONNX Runtime 1.31.1, CUDA EP | `.onnx` | 25.88 | 1.89 | PASS (cos ≥ 0.9999) |
43
+ | Core Ultra 9 386H CPU | ONNX Runtime 1.31.1, CPU EP | `.onnx` | 385.53 | 0.17 | reference |
44
+ | Snapdragon X2 Elite NPU | Qualcomm AI Hub, precompiled QNN ONNX | `.onnx` compiled by AI Hub | 76.93 | 0.45 | not compared (AI Hub profile job, 2026-10-30) |
45
+ | Snapdragon X2 Elite NPU | Qualcomm AI Hub, ONNX Runtime QNN EP | `.onnx`, compiled when it loads | 75.50 | 91.67 | not compared (AI Hub profile job, 2026-10-30) |
46
+ | Galaxy S26 NPU | Qualcomm AI Hub, precompiled QNN ONNX | `.onnx` compiled by AI Hub | 75.99 | 0.32 | not compared (AI Hub profile job, 2026-10-30) |
47
+
48
+ The ONNX file against the source `.tflite` (LiteRT interpreter on the CPU, the same input): PASS, cosine ≥ 0.9999 for every output.
49
+
50
+ ## Run it on the Intel NPU
51
+
52
+ Download this repo, then run the snippets from its folder:
53
+
54
+ ```sh
55
+ pip install huggingface_hub
56
+ hf download mlboydaisuke/Vision-RWKV-S-OpenVINO --local-dir Vision-RWKV-S-OpenVINO
57
+ cd Vision-RWKV-S-OpenVINO
58
+ ```
59
+
60
+ OpenVINO (`pip install openvino==2026.4.1`):
61
+
62
+ ```python
63
+ import numpy as np
64
+ import openvino as ov
65
+
66
+ core = ov.Core()
67
+ if "NPU" not in core.available_devices:
68
+ raise SystemExit(f"no Intel NPU visible to OpenVINO: {core.available_devices}")
69
+ with open("npu/vrwkv_s_fp16.blob", "rb") as f:
70
+ compiled = core.import_model(f.read(), "NPU") # precompiled: no compile at load
71
+ # or compile the IR on this machine: compiled = core.compile_model("openvino/vrwkv_s_fp16.xml", "NPU")
72
+ # replace the zeros with your preprocessed input (float32, the shapes printed by compiled.inputs)
73
+ feeds = {i.get_any_name(): np.zeros(list(i.shape), np.float32) for i in compiled.inputs}
74
+ outputs = compiled(feeds)
75
+ print({o.get_any_name(): outputs[o].shape for o in compiled.outputs})
76
+ ```
77
+
78
+ ONNX Runtime with the OpenVINO EP (`pip install onnxruntime==1.31.1 onnxruntime-ep-openvino==1.8.1`):
79
+
80
+ ```python
81
+ import numpy as np
82
+ import onnxruntime as ort
83
+ import onnxruntime_ep_openvino as ov_ep
84
+
85
+ ort.register_execution_provider_library(ov_ep.get_ep_name(), ov_ep.get_library_path())
86
+ npu = [d for d in ort.get_ep_devices()
87
+ if d.ep_name == ov_ep.get_ep_name() and d.ep_metadata.get("ov_device") == "NPU"]
88
+ if not npu:
89
+ raise SystemExit("no Intel NPU visible to the OpenVINO EP")
90
+ so = ort.SessionOptions()
91
+ so.add_provider_for_devices(npu, {})
92
+ sess = ort.InferenceSession("vrwkv_s_fp16.onnx", sess_options=so) # compiles for the NPU here
93
+ # replace the zeros with your preprocessed input (float32)
94
+ feeds = {i.name: np.zeros(i.shape, np.float32) for i in sess.get_inputs()}
95
+ outputs = sess.run(None, feeds)
96
+ print([o.shape for o in outputs])
97
+ ```
98
+
99
+ ## Licence
100
+
101
+ Apache-2.0, the licence of [litert-community/Vision-RWKV-S-LiteRT](https://huggingface.co/litert-community/Vision-RWKV-S-LiteRT). The upstream model: [OpenGVLab/Vision-RWKV](https://huggingface.co/OpenGVLab/Vision-RWKV).
manifest.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "source_repo": "litert-community/Vision-RWKV-S-LiteRT",
3
+ "source_revision": "184992fed3c7bb2761220287bcd0a496959e4e39",
4
+ "graphs": {
5
+ "vrwkv_s_fp16": {
6
+ "from": "vrwkv_s_fp16",
7
+ "ir": {
8
+ "openvino": "2026.4.1-22982-e213a147257-releases/2026/4",
9
+ "compress_to_fp16": false
10
+ },
11
+ "blob_compiled_with": {
12
+ "openvino": "2026.4.1-22982-e213a147257-releases/2026/4",
13
+ "FULL_DEVICE_NAME": "Intel(R) AI Boost",
14
+ "NPU_DRIVER_VERSION": "1004841",
15
+ "NPU_COMPILER_VERSION": "524292",
16
+ "NPU_COMPILER_TYPE": "CompilerType.PREFER_PLUGIN",
17
+ "INFERENCE_PRECISION_HINT": "<Type: 'float16'>"
18
+ },
19
+ "files": {
20
+ "vrwkv_s_fp16.onnx": {
21
+ "bytes": 95635055,
22
+ "sha256": "12184150736b7ec227a0a98ca6fa70ab69c8894ecb9cba800ce2537c6d75aaed"
23
+ },
24
+ "openvino/vrwkv_s_fp16.xml": {
25
+ "bytes": 1546499,
26
+ "sha256": "f34f101344f8e83faddd5c81987249a028057f284edd41057308e11907390d5e"
27
+ },
28
+ "openvino/vrwkv_s_fp16.bin": {
29
+ "bytes": 95336452,
30
+ "sha256": "54b97b3387a49b24689b5e78ad3d6004d78b742d327c30805b76a802ffa95782"
31
+ },
32
+ "npu/vrwkv_s_fp16.blob": {
33
+ "bytes": 74580104,
34
+ "sha256": "bed2c0e5fe4cc71621824d94642135ec77a1b0ca3bd384f861294a95bfaf76fe"
35
+ }
36
+ }
37
+ }
38
+ }
39
+ }
npu/vrwkv_s_fp16.blob ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bed2c0e5fe4cc71621824d94642135ec77a1b0ca3bd384f861294a95bfaf76fe
3
+ size 74580104
openvino/vrwkv_s_fp16.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54b97b3387a49b24689b5e78ad3d6004d78b742d327c30805b76a802ffa95782
3
+ size 95336452
openvino/vrwkv_s_fp16.xml ADDED
The diff for this file is too large to render. See raw diff
 
vrwkv_s_fp16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:12184150736b7ec227a0a98ca6fa70ab69c8894ecb9cba800ce2537c6d75aaed
3
+ size 95635055