{ "model": "depth_v2s", "upstream": { "model": "depth-anything/Depth-Anything-V2-Small-hf", "weights": { "repo": "depth-anything/Depth-Anything-V2-Small-hf", "revision": "5426e4f0f36572d16453bbda7a8389317b1bef99", "official": true }, "license": "Apache-2.0" }, "versions": { "python": "3.12.15+meta", "torch": "2.14.1+cpu", "torchao": "0.18.0+cpu", "executorch": "1.5.1+cpu", "executorch_git": "3b60683923245cf472b7323426920e15623ba361", "transformers": "5.19.0", "huggingface_hub": "2.2.0" }, "export": { "api": "torch.export.export(strict=False) -> to_edge_transform_and_lower(partitioner=[WebGPUPartitioner()]) -> to_executorch()", "partitioner": "executorch.backends.webgpu.partitioner.WebGPUPartitioner() (default compile options; wraps VulkanPartitioner)", "inputs": [ { "shape": [ 1, 3, 224, 224 ], "dtype": "float32", "layout": "NCHW" } ], "outputs": [ { "shape": [ 1, 224, 224 ], "dtype": "float32" } ], "dtype": "float32", "quantization": "none (fp32)", "attn_implementation": "eager", "graph_rewrites": [ "transformers 5.x all-zero bidirectional attention mask skipped (attention_mask=None; no padding in images)", "DINOv2 position embeddings resized (bicubic, as upstream) for the fixed input size in eager and stored as a constant" ], "preprocessing_outside_graph": "RGB / 255, ImageNet mean/std normalization, resize to input size", "postprocessing_outside_graph": "min-max normalize predicted_depth for display" }, "files": { "depth_v2s_webgpu.pte": { "bytes": 97307268, "sha256": "285602ed3a34a26a4d9d36355ce978587ed555e68b6b5fa8a567c1155a3b5160" }, "depth_v2s_portable.pte": { "bytes": 97290408, "sha256": "ce840e1c17ea65fcc52b7d8912dfe2d4d598350e3bdf6b8150f3d8c05d18423c", "note": "CPU-only validation artifact (portable kernels, no delegate); not for release" } }, "graph": { "non_delegated_kernels": {}, "delegates": [ { "id": "VulkanBackend", "ops": { "et_vk.prepack.default": 39, "aten.convolution.default": 24, "aten.view_copy.default": 113, "aten.permute_copy.default": 65, "aten.clone.default": 7, "aten.cat.default": 1, "aten.add.Tensor": 35, "aten.native_layer_norm.default": 28, "aten.squeeze_copy.dims": 97, "aten.linear.default": 72, "aten.bmm.default": 24, "aten.mul.Tensor": 37, "aten._softmax.default": 12, "aten.gelu.default": 12, "aten.slice_copy.Tensor": 4, "aten.relu.default": 7, "et_vk.conv_with_clamp.default": 9, "aten.upsample_bilinear2d.vec": 5, "dim_order_ops._to_dim_order_copy.default": 1 }, "inputs": 1, "outputs": 1 } ] }, "command": "python recipes/depth-anything-v2-small.py --out out/depth-anything-v2-small" }