Add ExecuTorch WebGPU program for modnet
Browse files
README.md
CHANGED
|
@@ -13,15 +13,13 @@ tags:
|
|
| 13 |
This repository contains an [ExecuTorch](https://github.com/pytorch/executorch) conversion of
|
| 14 |
https://github.com/ZHKKKe/MODNet for the ExecuTorch **WebGPU** backend, so the model runs entirely in a web browser on
|
| 15 |
the visitor's GPU. It powers the corresponding demo in the
|
| 16 |
-
|
| 17 |
|
| 18 |
- **Base model:** https://github.com/ZHKKKe/MODNet
|
| 19 |
- **Format:** ExecuTorch `.pte` programs delegated to the WebGPU backend
|
| 20 |
- **Quantization:** fp32 (as exported)
|
| 21 |
- **Total size:** 24.9 MB
|
| 22 |
|
| 23 |
-
> **Note:** Confirm that the pretrained portrait weights carry the repository's Apache-2.0 terms.
|
| 24 |
-
|
| 25 |
## Files
|
| 26 |
|
| 27 |
| File | Size | SHA-256 |
|
|
@@ -34,12 +32,12 @@ Full checksums are in `manifest.json`. The playground verifies them after downlo
|
|
| 34 |
|
| 35 |
These programs need the ExecuTorch WebGPU runtime compiled to WebAssembly
|
| 36 |
([backends/webgpu](https://github.com/pytorch/executorch/tree/main/backends/webgpu)). The easiest way to
|
| 37 |
-
try the model is the
|
| 38 |
-
files once, caches them in the browser, and runs everything locally.
|
| 39 |
|
| 40 |
## Provenance
|
| 41 |
|
| 42 |
-
- **Weights:** `boopathiraj/MODNet` @ `a967175f8fd63de67a1c161d8daac966d2b57e3a`, file `modnet/pretrained/modnet_photographic_portrait_matting.ckpt`, sha256 `7c22235f0925deba15d4d63e53afcb654c47055bbcd98f56e393ab2584007ed8` (community mirror
|
| 43 |
- **Exported with:** executorch 1.5.1+cpu, torch 2.14.1+cpu, torchao 0.18.0+cpu, transformers 5.19.0 (executorch git `3b6068392324`)
|
| 44 |
- **Reproduce:** `python recipes/modnet.py --out out/modnet` (recipe in `export/`)
|
| 45 |
- Re-running the recipe produces byte-identical files.
|
|
|
|
| 13 |
This repository contains an [ExecuTorch](https://github.com/pytorch/executorch) conversion of
|
| 14 |
https://github.com/ZHKKKe/MODNet for the ExecuTorch **WebGPU** backend, so the model runs entirely in a web browser on
|
| 15 |
the visitor's GPU. It powers the corresponding demo in the
|
| 16 |
+
ExecuTorch WebGPU Playground (launching soon on [executorch.ai](https://executorch.ai)).
|
| 17 |
|
| 18 |
- **Base model:** https://github.com/ZHKKKe/MODNet
|
| 19 |
- **Format:** ExecuTorch `.pte` programs delegated to the WebGPU backend
|
| 20 |
- **Quantization:** fp32 (as exported)
|
| 21 |
- **Total size:** 24.9 MB
|
| 22 |
|
|
|
|
|
|
|
| 23 |
## Files
|
| 24 |
|
| 25 |
| File | Size | SHA-256 |
|
|
|
|
| 32 |
|
| 33 |
These programs need the ExecuTorch WebGPU runtime compiled to WebAssembly
|
| 34 |
([backends/webgpu](https://github.com/pytorch/executorch/tree/main/backends/webgpu)). The easiest way to
|
| 35 |
+
try the model is the ExecuTorch WebGPU Playground (launching soon on [executorch.ai](https://executorch.ai)),
|
| 36 |
+
which downloads these files once, caches them in the browser, and runs everything locally.
|
| 37 |
|
| 38 |
## Provenance
|
| 39 |
|
| 40 |
+
- **Weights:** `boopathiraj/MODNet` @ `a967175f8fd63de67a1c161d8daac966d2b57e3a`, file `modnet/pretrained/modnet_photographic_portrait_matting.ckpt`, sha256 `7c22235f0925deba15d4d63e53afcb654c47055bbcd98f56e393ab2584007ed8` (from a community mirror of the official release)
|
| 41 |
- **Exported with:** executorch 1.5.1+cpu, torch 2.14.1+cpu, torchao 0.18.0+cpu, transformers 5.19.0 (executorch git `3b6068392324`)
|
| 42 |
- **Reproduce:** `python recipes/modnet.py --out out/modnet` (recipe in `export/`)
|
| 43 |
- Re-running the recipe produces byte-identical files.
|