| { |
| "schema": "trellis-webgpu-pack", |
| "version": 1, |
| "stage": "litoRgbaEncoder", |
| "upstreamRevision": "apple/ml-lito", |
| "shards": [ |
| { |
| "file": "stage-000.bin", |
| "bytes": 1617920, |
| "sha256": "c1b793c46b970a5e8cfac71e1854bc0d73973fa7357390294f39b80beb30a1ad" |
| } |
| ], |
| "tensors": { |
| "learnable_model.bias": { |
| "shape": [ |
| 1024 |
| ], |
| "encoding": "raw", |
| "dtype": "f16", |
| "data": { |
| "shard": 0, |
| "offset": 0, |
| "bytes": 2048 |
| } |
| }, |
| "learnable_model.weight": { |
| "shape": [ |
| 1024, |
| 4, |
| 14, |
| 14 |
| ], |
| "encoding": "raw", |
| "dtype": "f16", |
| "data": { |
| "shard": 0, |
| "offset": 2048, |
| "bytes": 1605632 |
| } |
| }, |
| "learnable_paddings": { |
| "shape": [ |
| 5, |
| 1024 |
| ], |
| "encoding": "raw", |
| "dtype": "f16", |
| "data": { |
| "shard": 0, |
| "offset": 1607680, |
| "bytes": 10240 |
| } |
| } |
| } |
| } |
|
|