Add ZipDepth Base ExecuTorch exports

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.pte filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,3 +1,48 @@
 ---
 license: mit
+pipeline_tag: depth-estimation
+library_name: executorch
 ---
+
+# zipdepth
+
+This repository hosts **ZipDepth Base NPU** (single-image depth estimation)
+exported for the
+[React Native ExecuTorch](https://www.npmjs.com/package/react-native-executorch)
+library as ExecuTorch `.pte` programs, ready to run on device.
+
+Upstream model: [fabiotosi92/ZipDepth](https://github.com/fabiotosi92/ZipDepth),
+checkpoint [`zipdepth_base_npu.pth`](https://github.com/fabiotosi92/ZipDepth/blob/91f3fd21e131641f51e8d35736d1958350180e3a/checkpoints/zipdepth_base_npu.pth),
+revision `91f3fd21e131641f51e8d35736d1958350180e3a`.
+
+## Variants
+
+| Path | Backend | Precision |
+| --- | --- | --- |
+| `base/coreml/zipdepth_base_384_coreml_fp32.pte` | coreml | fp32 |
+| `base/vulkan/zipdepth_base_384_vulkan_fp16.pte` | vulkan | fp16 |
+| `base/xnnpack/zipdepth_base_384_xnnpack_fp32.pte` | xnnpack | fp32 |
+
+## Repository structure
+
+```
+config.json
+base/coreml/config.json
+base/coreml/zipdepth_base_384_coreml_fp32.pte       24.9 MB
+base/vulkan/config.json
+base/vulkan/zipdepth_base_384_vulkan_fp16.pte       12.3 MB
+base/xnnpack/config.json
+base/xnnpack/zipdepth_base_384_xnnpack_fp32.pte     24.6 MB
+```
+
+## Compatibility
+
+These files are published for the **ExecuTorch v1.4.1** runtime. ExecuTorch
+gives no forward compatibility guarantee, so an older runtime may fail to load
+them.
+
+For React Native ExecuTorch model loading, see the
+[documentation](https://docs.swmansion.com/react-native-executorch/docs/fundamentals/downloading-models).
+To load these files in your own runtime, read the
+[compatibility note](https://github.com/pytorch/executorch/blob/main/runtime/COMPATIBILITY.md)
+first.

base/coreml/config.json

@@ -0,0 +1,43 @@
+{
+  "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
+  "model": "zipdepth_base",
+  "family": "zipdepth",
+  "capabilities": [
+    "depth-estimation"
+  ],
+  "backend": "coreml",
+  "license": "mit",
+  "variants": [
+    {
+      "file": "zipdepth_base_384_coreml_fp32.pte",
+      "precision": "fp32",
+      "quantized": false,
+      "default": true,
+      "methods": {
+        "forward": {
+          "inputs": [
+            {
+              "shape": [
+                1,
+                3,
+                384,
+                384
+              ],
+              "dtype": "float32"
+            }
+          ],
+          "outputs": [
+            {
+              "shape": [
+                1,
+                384,
+                384
+              ],
+              "dtype": "float32"
+            }
+          ]
+        }
+      }
+    }
+  ]
+}

base/coreml/zipdepth_base_384_coreml_fp32.pte

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9fc077311a72ac3b020423d28db47ec05e2599b578c6d14854ac321800b4cd90
+size 24858521

base/vulkan/config.json

@@ -0,0 +1,43 @@
+{
+  "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
+  "model": "zipdepth_base",
+  "family": "zipdepth",
+  "capabilities": [
+    "depth-estimation"
+  ],
+  "backend": "vulkan",
+  "license": "mit",
+  "variants": [
+    {
+      "file": "zipdepth_base_384_vulkan_fp16.pte",
+      "precision": "fp16",
+      "quantized": false,
+      "default": true,
+      "methods": {
+        "forward": {
+          "inputs": [
+            {
+              "shape": [
+                1,
+                3,
+                384,
+                384
+              ],
+              "dtype": "float32"
+            }
+          ],
+          "outputs": [
+            {
+              "shape": [
+                1,
+                384,
+                384
+              ],
+              "dtype": "float32"
+            }
+          ]
+        }
+      }
+    }
+  ]
+}

base/vulkan/zipdepth_base_384_vulkan_fp16.pte

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f3966cc629e1ba8f13d6fd7eb9a5abfe05df99c3f802cddd70fb428548b0e424
+size 12341250

base/xnnpack/config.json

@@ -0,0 +1,43 @@
+{
+  "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
+  "model": "zipdepth_base",
+  "family": "zipdepth",
+  "capabilities": [
+    "depth-estimation"
+  ],
+  "backend": "xnnpack",
+  "license": "mit",
+  "variants": [
+    {
+      "file": "zipdepth_base_384_xnnpack_fp32.pte",
+      "precision": "fp32",
+      "quantized": false,
+      "default": true,
+      "methods": {
+        "forward": {
+          "inputs": [
+            {
+              "shape": [
+                1,
+                3,
+                384,
+                384
+              ],
+              "dtype": "float32"
+            }
+          ],
+          "outputs": [
+            {
+              "shape": [
+                1,
+                384,
+                384
+              ],
+              "dtype": "float32"
+            }
+          ]
+        }
+      }
+    }
+  ]
+}

base/xnnpack/zipdepth_base_384_xnnpack_fp32.pte

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a373195cf37ebf7b041ff4c5e85f8680bb36744a2c9cfb349ba715ec8512d886
+size 24609344

config.json

@@ -0,0 +1,3 @@
+{
+  "modelName": "zipdepth"
+}