Upload 48 files
Browse files- CodeDecoder.mlmodelc/analytics/coremldata.bin +3 -0
- CodeDecoder.mlmodelc/coremldata.bin +3 -0
- CodeDecoder.mlmodelc/metadata.json +160 -0
- CodeDecoder.mlmodelc/model.mil +0 -0
- CodeDecoder.mlmodelc/weights/weight.bin +3 -0
- CodeDecoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- CodeDecoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- CodeDecoder.mlpackage/Manifest.json +18 -0
- CodeEmbedder.mlmodelc/analytics/coremldata.bin +3 -0
- CodeEmbedder.mlmodelc/coremldata.bin +3 -0
- CodeEmbedder.mlmodelc/metadata.json +66 -0
- CodeEmbedder.mlmodelc/model.mil +26 -0
- CodeEmbedder.mlmodelc/weights/weight.bin +3 -0
- CodeEmbedder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- CodeEmbedder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- CodeEmbedder.mlpackage/Manifest.json +18 -0
- MultiCodeDecoder.mlmodelc/analytics/coremldata.bin +3 -0
- MultiCodeDecoder.mlmodelc/coremldata.bin +3 -0
- MultiCodeDecoder.mlmodelc/metadata.json +162 -0
- MultiCodeDecoder.mlmodelc/model.mil +0 -0
- MultiCodeDecoder.mlmodelc/weights/weight.bin +3 -0
- MultiCodeDecoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- MultiCodeDecoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- MultiCodeDecoder.mlpackage/Manifest.json +18 -0
- MultiCodeEmbedder.mlmodelc/analytics/coremldata.bin +3 -0
- MultiCodeEmbedder.mlmodelc/coremldata.bin +3 -0
- MultiCodeEmbedder.mlmodelc/metadata.json +66 -0
- MultiCodeEmbedder.mlmodelc/model.mil +26 -0
- MultiCodeEmbedder.mlmodelc/weights/weight.bin +3 -0
- MultiCodeEmbedder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- MultiCodeEmbedder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- MultiCodeEmbedder.mlpackage/Manifest.json +18 -0
- SpeechDecoder.mlmodelc/analytics/coremldata.bin +3 -0
- SpeechDecoder.mlmodelc/coremldata.bin +3 -0
- SpeechDecoder.mlmodelc/metadata.json +85 -0
- SpeechDecoder.mlmodelc/model.mil +0 -0
- SpeechDecoder.mlmodelc/weights/weight.bin +3 -0
- SpeechDecoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- SpeechDecoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- SpeechDecoder.mlpackage/Manifest.json +18 -0
- TextProjector.mlmodelc/analytics/coremldata.bin +3 -0
- TextProjector.mlmodelc/coremldata.bin +3 -0
- TextProjector.mlmodelc/metadata.json +68 -0
- TextProjector.mlmodelc/model.mil +31 -0
- TextProjector.mlmodelc/weights/weight.bin +3 -0
- TextProjector.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- TextProjector.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- TextProjector.mlpackage/Manifest.json +18 -0
CodeDecoder.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8345486c7d0ea287283e4e0194e3f53791882866168b6751dc649052d41503d8
|
| 3 |
+
size 243
|
CodeDecoder.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9078227add9378b790d09d263184a3481dee955ad6e23c5ee7aeacb9b9697aaa
|
| 3 |
+
size 641
|
CodeDecoder.mlmodelc/metadata.json
ADDED
|
@@ -0,0 +1,160 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"metadataOutputVersion" : "3.0",
|
| 4 |
+
"storagePrecision" : "Float32",
|
| 5 |
+
"outputSchema" : [
|
| 6 |
+
{
|
| 7 |
+
"hasShapeFlexibility" : "0",
|
| 8 |
+
"isOptional" : "0",
|
| 9 |
+
"dataType" : "Float16",
|
| 10 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 3072)",
|
| 11 |
+
"shortDescription" : "",
|
| 12 |
+
"shape" : "[1, 1, 3072]",
|
| 13 |
+
"name" : "logits",
|
| 14 |
+
"type" : "MultiArray"
|
| 15 |
+
},
|
| 16 |
+
{
|
| 17 |
+
"hasShapeFlexibility" : "0",
|
| 18 |
+
"isOptional" : "0",
|
| 19 |
+
"dataType" : "Float16",
|
| 20 |
+
"formattedType" : "MultiArray (Float16 1 × 1024 × 1 × 1)",
|
| 21 |
+
"shortDescription" : "",
|
| 22 |
+
"shape" : "[1, 1024, 1, 1]",
|
| 23 |
+
"name" : "hidden_states",
|
| 24 |
+
"type" : "MultiArray"
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"hasShapeFlexibility" : "0",
|
| 28 |
+
"isOptional" : "0",
|
| 29 |
+
"dataType" : "Float16",
|
| 30 |
+
"formattedType" : "MultiArray (Float16 1 × 28672 × 1 × 256)",
|
| 31 |
+
"shortDescription" : "",
|
| 32 |
+
"shape" : "[1, 28672, 1, 256]",
|
| 33 |
+
"name" : "new_key_cache",
|
| 34 |
+
"type" : "MultiArray"
|
| 35 |
+
},
|
| 36 |
+
{
|
| 37 |
+
"hasShapeFlexibility" : "0",
|
| 38 |
+
"isOptional" : "0",
|
| 39 |
+
"dataType" : "Float16",
|
| 40 |
+
"formattedType" : "MultiArray (Float16 1 × 28672 × 1 × 256)",
|
| 41 |
+
"shortDescription" : "",
|
| 42 |
+
"shape" : "[1, 28672, 1, 256]",
|
| 43 |
+
"name" : "new_value_cache",
|
| 44 |
+
"type" : "MultiArray"
|
| 45 |
+
}
|
| 46 |
+
],
|
| 47 |
+
"modelParameters" : [
|
| 48 |
+
|
| 49 |
+
],
|
| 50 |
+
"specificationVersion" : 9,
|
| 51 |
+
"mlProgramOperationTypeHistogram" : {
|
| 52 |
+
"Ios18.expandDims" : 235,
|
| 53 |
+
"Ios18.softmax" : 28,
|
| 54 |
+
"Ios18.cos" : 1,
|
| 55 |
+
"Ios18.mul" : 566,
|
| 56 |
+
"Ios18.matmul" : 57,
|
| 57 |
+
"Ios18.rsqrt" : 113,
|
| 58 |
+
"Ios16.reduceMean" : 113,
|
| 59 |
+
"Ios18.sin" : 1,
|
| 60 |
+
"Tile" : 56,
|
| 61 |
+
"Ios18.add" : 311,
|
| 62 |
+
"Ios18.reshape" : 280,
|
| 63 |
+
"Ios18.linear" : 197,
|
| 64 |
+
"Ios18.concat" : 59,
|
| 65 |
+
"Ios18.transpose" : 113,
|
| 66 |
+
"Ios18.sub" : 1,
|
| 67 |
+
"Ios18.pow" : 113,
|
| 68 |
+
"Ios18.cast" : 10,
|
| 69 |
+
"Ios18.silu" : 28,
|
| 70 |
+
"Ios18.sliceByIndex" : 168,
|
| 71 |
+
"Ios18.squeeze" : 283
|
| 72 |
+
},
|
| 73 |
+
"computePrecision" : "Mixed (Float16, Float32, Int32)",
|
| 74 |
+
"isUpdatable" : "0",
|
| 75 |
+
"stateSchema" : [
|
| 76 |
+
|
| 77 |
+
],
|
| 78 |
+
"availability" : {
|
| 79 |
+
"macOS" : "15.0",
|
| 80 |
+
"tvOS" : "18.0",
|
| 81 |
+
"visionOS" : "2.0",
|
| 82 |
+
"watchOS" : "11.0",
|
| 83 |
+
"iOS" : "18.0",
|
| 84 |
+
"macCatalyst" : "18.0"
|
| 85 |
+
},
|
| 86 |
+
"modelType" : {
|
| 87 |
+
"name" : "MLModelType_mlProgram"
|
| 88 |
+
},
|
| 89 |
+
"userDefinedMetadata" : {
|
| 90 |
+
"com.github.apple.coremltools.conversion_date" : "2026-03-22",
|
| 91 |
+
"com.github.apple.coremltools.source" : "torch==2.7.0",
|
| 92 |
+
"com.github.apple.coremltools.version" : "9.0",
|
| 93 |
+
"com.github.apple.coremltools.source_dialect" : "TorchScript"
|
| 94 |
+
},
|
| 95 |
+
"inputSchema" : [
|
| 96 |
+
{
|
| 97 |
+
"hasShapeFlexibility" : "0",
|
| 98 |
+
"isOptional" : "0",
|
| 99 |
+
"dataType" : "Float16",
|
| 100 |
+
"formattedType" : "MultiArray (Float16 1 × 1024 × 1 × 1)",
|
| 101 |
+
"shortDescription" : "",
|
| 102 |
+
"shape" : "[1, 1024, 1, 1]",
|
| 103 |
+
"name" : "input_embeds",
|
| 104 |
+
"type" : "MultiArray"
|
| 105 |
+
},
|
| 106 |
+
{
|
| 107 |
+
"hasShapeFlexibility" : "0",
|
| 108 |
+
"isOptional" : "0",
|
| 109 |
+
"dataType" : "Int32",
|
| 110 |
+
"formattedType" : "MultiArray (Int32 1)",
|
| 111 |
+
"shortDescription" : "",
|
| 112 |
+
"shape" : "[1]",
|
| 113 |
+
"name" : "cache_length",
|
| 114 |
+
"type" : "MultiArray"
|
| 115 |
+
},
|
| 116 |
+
{
|
| 117 |
+
"hasShapeFlexibility" : "0",
|
| 118 |
+
"isOptional" : "0",
|
| 119 |
+
"dataType" : "Float16",
|
| 120 |
+
"formattedType" : "MultiArray (Float16 1 × 256)",
|
| 121 |
+
"shortDescription" : "",
|
| 122 |
+
"shape" : "[1, 256]",
|
| 123 |
+
"name" : "key_padding_mask",
|
| 124 |
+
"type" : "MultiArray"
|
| 125 |
+
},
|
| 126 |
+
{
|
| 127 |
+
"hasShapeFlexibility" : "0",
|
| 128 |
+
"isOptional" : "0",
|
| 129 |
+
"dataType" : "Float16",
|
| 130 |
+
"formattedType" : "MultiArray (Float16 1 × 256)",
|
| 131 |
+
"shortDescription" : "",
|
| 132 |
+
"shape" : "[1, 256]",
|
| 133 |
+
"name" : "kv_cache_update_mask",
|
| 134 |
+
"type" : "MultiArray"
|
| 135 |
+
},
|
| 136 |
+
{
|
| 137 |
+
"hasShapeFlexibility" : "0",
|
| 138 |
+
"isOptional" : "0",
|
| 139 |
+
"dataType" : "Float16",
|
| 140 |
+
"formattedType" : "MultiArray (Float16 1 × 28672 × 1 × 256)",
|
| 141 |
+
"shortDescription" : "",
|
| 142 |
+
"shape" : "[1, 28672, 1, 256]",
|
| 143 |
+
"name" : "key_cache",
|
| 144 |
+
"type" : "MultiArray"
|
| 145 |
+
},
|
| 146 |
+
{
|
| 147 |
+
"hasShapeFlexibility" : "0",
|
| 148 |
+
"isOptional" : "0",
|
| 149 |
+
"dataType" : "Float16",
|
| 150 |
+
"formattedType" : "MultiArray (Float16 1 × 28672 × 1 × 256)",
|
| 151 |
+
"shortDescription" : "",
|
| 152 |
+
"shape" : "[1, 28672, 1, 256]",
|
| 153 |
+
"name" : "value_cache",
|
| 154 |
+
"type" : "MultiArray"
|
| 155 |
+
}
|
| 156 |
+
],
|
| 157 |
+
"generatedClassName" : "CodeDecoder",
|
| 158 |
+
"method" : "predict"
|
| 159 |
+
}
|
| 160 |
+
]
|
CodeDecoder.mlmodelc/model.mil
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
CodeDecoder.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ac65c785ed811f32ecf104b7eb5964e8337ae25e69e3b9388d9e59f728862da3
|
| 3 |
+
size 1774497728
|
CodeDecoder.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b6dd851437cbd6bf5099d35a23a4dbde2260da1fb105486a2bd93a95f1e62e6b
|
| 3 |
+
size 715988
|
CodeDecoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ac65c785ed811f32ecf104b7eb5964e8337ae25e69e3b9388d9e59f728862da3
|
| 3 |
+
size 1774497728
|
CodeDecoder.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"0263A430-7CA4-4CFE-AF72-3450E3091735": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Specification",
|
| 7 |
+
"name": "model.mlmodel",
|
| 8 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 9 |
+
},
|
| 10 |
+
"FB7E765B-CAED-45B8-8D62-E03F08BEC5BC": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Weights",
|
| 13 |
+
"name": "weights",
|
| 14 |
+
"path": "com.apple.CoreML/weights"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "0263A430-7CA4-4CFE-AF72-3450E3091735"
|
| 18 |
+
}
|
CodeEmbedder.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f17e7f736e831b85b54889353af91b6cb0d7624561471bc2d295c1128169712a
|
| 3 |
+
size 243
|
CodeEmbedder.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1b2b5bc8ef42b5aed93e5c070c9cf7bf8c2c2409824b46b64060008f3d520438
|
| 3 |
+
size 381
|
CodeEmbedder.mlmodelc/metadata.json
ADDED
|
@@ -0,0 +1,66 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"metadataOutputVersion" : "3.0",
|
| 4 |
+
"storagePrecision" : "Float16",
|
| 5 |
+
"outputSchema" : [
|
| 6 |
+
{
|
| 7 |
+
"hasShapeFlexibility" : "0",
|
| 8 |
+
"isOptional" : "0",
|
| 9 |
+
"dataType" : "Float16",
|
| 10 |
+
"formattedType" : "MultiArray (Float16 1 × 1024 × 1 × 1)",
|
| 11 |
+
"shortDescription" : "",
|
| 12 |
+
"shape" : "[1, 1024, 1, 1]",
|
| 13 |
+
"name" : "input_embeds",
|
| 14 |
+
"type" : "MultiArray"
|
| 15 |
+
}
|
| 16 |
+
],
|
| 17 |
+
"modelParameters" : [
|
| 18 |
+
|
| 19 |
+
],
|
| 20 |
+
"specificationVersion" : 9,
|
| 21 |
+
"mlProgramOperationTypeHistogram" : {
|
| 22 |
+
"Ios18.greaterEqual" : 1,
|
| 23 |
+
"Ios18.add" : 1,
|
| 24 |
+
"Ios18.cast" : 3,
|
| 25 |
+
"Select" : 1,
|
| 26 |
+
"Ios18.gather" : 1,
|
| 27 |
+
"Ios18.expandDims" : 2
|
| 28 |
+
},
|
| 29 |
+
"computePrecision" : "Mixed (Float16, Int16, Int32)",
|
| 30 |
+
"isUpdatable" : "0",
|
| 31 |
+
"stateSchema" : [
|
| 32 |
+
|
| 33 |
+
],
|
| 34 |
+
"availability" : {
|
| 35 |
+
"macOS" : "15.0",
|
| 36 |
+
"tvOS" : "18.0",
|
| 37 |
+
"visionOS" : "2.0",
|
| 38 |
+
"watchOS" : "11.0",
|
| 39 |
+
"iOS" : "18.0",
|
| 40 |
+
"macCatalyst" : "18.0"
|
| 41 |
+
},
|
| 42 |
+
"modelType" : {
|
| 43 |
+
"name" : "MLModelType_mlProgram"
|
| 44 |
+
},
|
| 45 |
+
"userDefinedMetadata" : {
|
| 46 |
+
"com.github.apple.coremltools.conversion_date" : "2026-03-22",
|
| 47 |
+
"com.github.apple.coremltools.source" : "torch==2.10.0",
|
| 48 |
+
"com.github.apple.coremltools.version" : "9.0",
|
| 49 |
+
"com.github.apple.coremltools.source_dialect" : "TorchScript"
|
| 50 |
+
},
|
| 51 |
+
"inputSchema" : [
|
| 52 |
+
{
|
| 53 |
+
"hasShapeFlexibility" : "0",
|
| 54 |
+
"isOptional" : "0",
|
| 55 |
+
"dataType" : "Int32",
|
| 56 |
+
"formattedType" : "MultiArray (Int32 1)",
|
| 57 |
+
"shortDescription" : "",
|
| 58 |
+
"shape" : "[1]",
|
| 59 |
+
"name" : "input_ids",
|
| 60 |
+
"type" : "MultiArray"
|
| 61 |
+
}
|
| 62 |
+
],
|
| 63 |
+
"generatedClassName" : "CodeEmbedder",
|
| 64 |
+
"method" : "predict"
|
| 65 |
+
}
|
| 66 |
+
]
|
CodeEmbedder.mlmodelc/model.mil
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
program(1.3)
|
| 2 |
+
[buildInfo = dict<string, string>({{"coremlc-component-MIL", "3510.2.1"}, {"coremlc-version", "3500.32.1"}, {"coremltools-component-torch", "2.10.0"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})]
|
| 3 |
+
{
|
| 4 |
+
func main<ios18>(tensor<int32, [1]> input_ids) {
|
| 5 |
+
int32 embed_batch_dims_0 = const()[name = string("embed_batch_dims_0"), val = int32(0)];
|
| 6 |
+
bool embed_validate_indices_0 = const()[name = string("embed_validate_indices_0"), val = bool(false)];
|
| 7 |
+
tensor<fp16, [3072, 1024]> codec_embedding_weight_to_fp16 = const()[name = string("codec_embedding_weight_to_fp16"), val = tensor<fp16, [3072, 1024]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
|
| 8 |
+
string input_ids_to_int16_dtype_0 = const()[name = string("input_ids_to_int16_dtype_0"), val = string("int16")];
|
| 9 |
+
string cast_2_dtype_0 = const()[name = string("cast_2_dtype_0"), val = string("int32")];
|
| 10 |
+
int32 greater_equal_0_y_0 = const()[name = string("greater_equal_0_y_0"), val = int32(0)];
|
| 11 |
+
tensor<int16, [1]> input_ids_to_int16 = cast(dtype = input_ids_to_int16_dtype_0, x = input_ids)[name = string("cast_5")];
|
| 12 |
+
tensor<int32, [1]> cast_2 = cast(dtype = cast_2_dtype_0, x = input_ids_to_int16)[name = string("cast_4")];
|
| 13 |
+
tensor<bool, [1]> greater_equal_0 = greater_equal(x = cast_2, y = greater_equal_0_y_0)[name = string("greater_equal_0")];
|
| 14 |
+
int32 slice_by_index_0 = const()[name = string("slice_by_index_0"), val = int32(3072)];
|
| 15 |
+
tensor<int32, [1]> add_0 = add(x = cast_2, y = slice_by_index_0)[name = string("add_0")];
|
| 16 |
+
tensor<int32, [1]> select_0 = select(a = cast_2, b = add_0, cond = greater_equal_0)[name = string("select_0")];
|
| 17 |
+
int32 embed_cast_fp16_cast_uint16_axis_0 = const()[name = string("embed_cast_fp16_cast_uint16_axis_0"), val = int32(0)];
|
| 18 |
+
string select_0_to_int16_dtype_0 = const()[name = string("select_0_to_int16_dtype_0"), val = string("int16")];
|
| 19 |
+
tensor<int16, [1]> select_0_to_int16 = cast(dtype = select_0_to_int16_dtype_0, x = select_0)[name = string("cast_3")];
|
| 20 |
+
tensor<fp16, [1, 1024]> embed_cast_fp16_cast_uint16_cast_uint16 = gather(axis = embed_cast_fp16_cast_uint16_axis_0, batch_dims = embed_batch_dims_0, indices = select_0_to_int16, validate_indices = embed_validate_indices_0, x = codec_embedding_weight_to_fp16)[name = string("embed_cast_fp16_cast_uint16_cast_uint16")];
|
| 21 |
+
tensor<int32, [1]> var_8_axes_0 = const()[name = string("op_8_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 22 |
+
tensor<fp16, [1, 1024, 1]> var_8_cast_fp16 = expand_dims(axes = var_8_axes_0, x = embed_cast_fp16_cast_uint16_cast_uint16)[name = string("op_8_cast_fp16")];
|
| 23 |
+
tensor<int32, [1]> var_10_axes_0 = const()[name = string("op_10_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 24 |
+
tensor<fp16, [1, 1024, 1, 1]> input_embeds = expand_dims(axes = var_10_axes_0, x = var_8_cast_fp16)[name = string("op_10_cast_fp16")];
|
| 25 |
+
} -> (input_embeds);
|
| 26 |
+
}
|
CodeEmbedder.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dc523d30a88406fc4988b3e523cddba01b004a688071591be1db81e3abe0830b
|
| 3 |
+
size 6291584
|
CodeEmbedder.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bd76a225dbbae9754c93805c3377def520e00453108b527eb01eb93d135cf6fd
|
| 3 |
+
size 3366
|
CodeEmbedder.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dc523d30a88406fc4988b3e523cddba01b004a688071591be1db81e3abe0830b
|
| 3 |
+
size 6291584
|
CodeEmbedder.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"10EEEF0C-0099-4C00-A391-1F79A162CD55": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"72910515-6A65-44CD-BE0E-E7AB46883288": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Specification",
|
| 13 |
+
"name": "model.mlmodel",
|
| 14 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "72910515-6A65-44CD-BE0E-E7AB46883288"
|
| 18 |
+
}
|
MultiCodeDecoder.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a73bc7058148e8d187486bafe7450c61afb6545ac9159af85344c299beb19eb0
|
| 3 |
+
size 243
|
MultiCodeDecoder.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f26a77a0b60707f29f4fbcdbe2f8176cb89bb92c14429eaaafbfceaa58e3c739
|
| 3 |
+
size 635
|
MultiCodeDecoder.mlmodelc/metadata.json
ADDED
|
@@ -0,0 +1,162 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"metadataOutputVersion" : "3.0",
|
| 4 |
+
"storagePrecision" : "Mixed (Float32, Palettized (8 bits), UInt8)",
|
| 5 |
+
"outputSchema" : [
|
| 6 |
+
{
|
| 7 |
+
"hasShapeFlexibility" : "0",
|
| 8 |
+
"isOptional" : "0",
|
| 9 |
+
"dataType" : "Float16",
|
| 10 |
+
"formattedType" : "MultiArray (Float16 1 × 15 × 2048)",
|
| 11 |
+
"shortDescription" : "",
|
| 12 |
+
"shape" : "[1, 15, 2048]",
|
| 13 |
+
"name" : "all_logits",
|
| 14 |
+
"type" : "MultiArray"
|
| 15 |
+
},
|
| 16 |
+
{
|
| 17 |
+
"hasShapeFlexibility" : "0",
|
| 18 |
+
"isOptional" : "0",
|
| 19 |
+
"dataType" : "Float16",
|
| 20 |
+
"formattedType" : "MultiArray (Float16 1 × 1024 × 1 × 1)",
|
| 21 |
+
"shortDescription" : "",
|
| 22 |
+
"shape" : "[1, 1024, 1, 1]",
|
| 23 |
+
"name" : "hidden_states",
|
| 24 |
+
"type" : "MultiArray"
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"hasShapeFlexibility" : "0",
|
| 28 |
+
"isOptional" : "0",
|
| 29 |
+
"dataType" : "Float16",
|
| 30 |
+
"formattedType" : "MultiArray (Float16 1 × 5120 × 1 × 16)",
|
| 31 |
+
"shortDescription" : "",
|
| 32 |
+
"shape" : "[1, 5120, 1, 16]",
|
| 33 |
+
"name" : "new_key_cache",
|
| 34 |
+
"type" : "MultiArray"
|
| 35 |
+
},
|
| 36 |
+
{
|
| 37 |
+
"hasShapeFlexibility" : "0",
|
| 38 |
+
"isOptional" : "0",
|
| 39 |
+
"dataType" : "Float16",
|
| 40 |
+
"formattedType" : "MultiArray (Float16 1 × 5120 × 1 × 16)",
|
| 41 |
+
"shortDescription" : "",
|
| 42 |
+
"shape" : "[1, 5120, 1, 16]",
|
| 43 |
+
"name" : "new_value_cache",
|
| 44 |
+
"type" : "MultiArray"
|
| 45 |
+
}
|
| 46 |
+
],
|
| 47 |
+
"modelParameters" : [
|
| 48 |
+
|
| 49 |
+
],
|
| 50 |
+
"specificationVersion" : 9,
|
| 51 |
+
"mlProgramOperationTypeHistogram" : {
|
| 52 |
+
"Ios18.expandDims" : 49,
|
| 53 |
+
"Ios18.softmax" : 5,
|
| 54 |
+
"Ios18.cos" : 1,
|
| 55 |
+
"Ios18.mul" : 106,
|
| 56 |
+
"Ios18.matmul" : 11,
|
| 57 |
+
"Ios18.rsqrt" : 21,
|
| 58 |
+
"Ios16.reduceMean" : 21,
|
| 59 |
+
"Ios18.sin" : 1,
|
| 60 |
+
"Tile" : 10,
|
| 61 |
+
"Ios18.add" : 58,
|
| 62 |
+
"Ios18.reshape" : 50,
|
| 63 |
+
"Ios18.constexprLutToDense" : 50,
|
| 64 |
+
"Ios18.linear" : 50,
|
| 65 |
+
"Ios18.concat" : 13,
|
| 66 |
+
"Ios18.transpose" : 21,
|
| 67 |
+
"Ios18.sub" : 1,
|
| 68 |
+
"Ios18.pow" : 21,
|
| 69 |
+
"Ios18.cast" : 10,
|
| 70 |
+
"Ios18.silu" : 5,
|
| 71 |
+
"Stack" : 1,
|
| 72 |
+
"Ios18.sliceByIndex" : 30,
|
| 73 |
+
"Ios18.squeeze" : 67
|
| 74 |
+
},
|
| 75 |
+
"computePrecision" : "Mixed (Float16, Float32, Int32)",
|
| 76 |
+
"isUpdatable" : "0",
|
| 77 |
+
"stateSchema" : [
|
| 78 |
+
|
| 79 |
+
],
|
| 80 |
+
"availability" : {
|
| 81 |
+
"macOS" : "15.0",
|
| 82 |
+
"tvOS" : "18.0",
|
| 83 |
+
"visionOS" : "2.0",
|
| 84 |
+
"watchOS" : "11.0",
|
| 85 |
+
"iOS" : "18.0",
|
| 86 |
+
"macCatalyst" : "18.0"
|
| 87 |
+
},
|
| 88 |
+
"modelType" : {
|
| 89 |
+
"name" : "MLModelType_mlProgram"
|
| 90 |
+
},
|
| 91 |
+
"userDefinedMetadata" : {
|
| 92 |
+
"com.github.apple.coremltools.conversion_date" : "2026-03-22",
|
| 93 |
+
"com.github.apple.coremltools.source" : "torch==2.7.0",
|
| 94 |
+
"com.github.apple.coremltools.version" : "9.0",
|
| 95 |
+
"com.github.apple.coremltools.source_dialect" : "TorchScript"
|
| 96 |
+
},
|
| 97 |
+
"inputSchema" : [
|
| 98 |
+
{
|
| 99 |
+
"hasShapeFlexibility" : "0",
|
| 100 |
+
"isOptional" : "0",
|
| 101 |
+
"dataType" : "Float16",
|
| 102 |
+
"formattedType" : "MultiArray (Float16 1 × 1024 × 1 × 1)",
|
| 103 |
+
"shortDescription" : "",
|
| 104 |
+
"shape" : "[1, 1024, 1, 1]",
|
| 105 |
+
"name" : "input_embeds",
|
| 106 |
+
"type" : "MultiArray"
|
| 107 |
+
},
|
| 108 |
+
{
|
| 109 |
+
"hasShapeFlexibility" : "0",
|
| 110 |
+
"isOptional" : "0",
|
| 111 |
+
"dataType" : "Int32",
|
| 112 |
+
"formattedType" : "MultiArray (Int32 1)",
|
| 113 |
+
"shortDescription" : "",
|
| 114 |
+
"shape" : "[1]",
|
| 115 |
+
"name" : "cache_length",
|
| 116 |
+
"type" : "MultiArray"
|
| 117 |
+
},
|
| 118 |
+
{
|
| 119 |
+
"hasShapeFlexibility" : "0",
|
| 120 |
+
"isOptional" : "0",
|
| 121 |
+
"dataType" : "Float16",
|
| 122 |
+
"formattedType" : "MultiArray (Float16 1 × 5120 × 1 × 16)",
|
| 123 |
+
"shortDescription" : "",
|
| 124 |
+
"shape" : "[1, 5120, 1, 16]",
|
| 125 |
+
"name" : "key_cache",
|
| 126 |
+
"type" : "MultiArray"
|
| 127 |
+
},
|
| 128 |
+
{
|
| 129 |
+
"hasShapeFlexibility" : "0",
|
| 130 |
+
"isOptional" : "0",
|
| 131 |
+
"dataType" : "Float16",
|
| 132 |
+
"formattedType" : "MultiArray (Float16 1 × 16)",
|
| 133 |
+
"shortDescription" : "",
|
| 134 |
+
"shape" : "[1, 16]",
|
| 135 |
+
"name" : "key_padding_mask",
|
| 136 |
+
"type" : "MultiArray"
|
| 137 |
+
},
|
| 138 |
+
{
|
| 139 |
+
"hasShapeFlexibility" : "0",
|
| 140 |
+
"isOptional" : "0",
|
| 141 |
+
"dataType" : "Float16",
|
| 142 |
+
"formattedType" : "MultiArray (Float16 1 × 16)",
|
| 143 |
+
"shortDescription" : "",
|
| 144 |
+
"shape" : "[1, 16]",
|
| 145 |
+
"name" : "kv_cache_update_mask",
|
| 146 |
+
"type" : "MultiArray"
|
| 147 |
+
},
|
| 148 |
+
{
|
| 149 |
+
"hasShapeFlexibility" : "0",
|
| 150 |
+
"isOptional" : "0",
|
| 151 |
+
"dataType" : "Float16",
|
| 152 |
+
"formattedType" : "MultiArray (Float16 1 × 5120 × 1 × 16)",
|
| 153 |
+
"shortDescription" : "",
|
| 154 |
+
"shape" : "[1, 5120, 1, 16]",
|
| 155 |
+
"name" : "value_cache",
|
| 156 |
+
"type" : "MultiArray"
|
| 157 |
+
}
|
| 158 |
+
],
|
| 159 |
+
"generatedClassName" : "MultiCodeDecoder",
|
| 160 |
+
"method" : "predict"
|
| 161 |
+
}
|
| 162 |
+
]
|
MultiCodeDecoder.mlmodelc/model.mil
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
MultiCodeDecoder.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2bed887bd321e162bf9993197bcc882d7fee120a4ffcac638452844b3cddbf6f
|
| 3 |
+
size 110234752
|
MultiCodeDecoder.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:21e3723b1f66d3a237c5462e7801ff96a7e8eaeaa0b00086ca83b6041ceade9d
|
| 3 |
+
size 150675
|
MultiCodeDecoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2bed887bd321e162bf9993197bcc882d7fee120a4ffcac638452844b3cddbf6f
|
| 3 |
+
size 110234752
|
MultiCodeDecoder.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"ABCF1C61-2114-489E-9BCC-231CDBEE037D": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"C2DBFC38-9470-4E00-B0F2-94A4154EBABA": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Specification",
|
| 13 |
+
"name": "model.mlmodel",
|
| 14 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "C2DBFC38-9470-4E00-B0F2-94A4154EBABA"
|
| 18 |
+
}
|
MultiCodeEmbedder.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:44c4ed23aead0da950d21abe61c62c67151710f0dd2cb231e18ecc25d33fd882
|
| 3 |
+
size 243
|
MultiCodeEmbedder.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f5f84d6adb86a8ca20fd9991f6f3e37e00af10c957150222967069ba7727719d
|
| 3 |
+
size 381
|
MultiCodeEmbedder.mlmodelc/metadata.json
ADDED
|
@@ -0,0 +1,66 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"metadataOutputVersion" : "3.0",
|
| 4 |
+
"storagePrecision" : "Float16",
|
| 5 |
+
"outputSchema" : [
|
| 6 |
+
{
|
| 7 |
+
"hasShapeFlexibility" : "0",
|
| 8 |
+
"isOptional" : "0",
|
| 9 |
+
"dataType" : "Float16",
|
| 10 |
+
"formattedType" : "MultiArray (Float16 1 × 1024 × 1 × 1)",
|
| 11 |
+
"shortDescription" : "",
|
| 12 |
+
"shape" : "[1, 1024, 1, 1]",
|
| 13 |
+
"name" : "input_embeds",
|
| 14 |
+
"type" : "MultiArray"
|
| 15 |
+
}
|
| 16 |
+
],
|
| 17 |
+
"modelParameters" : [
|
| 18 |
+
|
| 19 |
+
],
|
| 20 |
+
"specificationVersion" : 9,
|
| 21 |
+
"mlProgramOperationTypeHistogram" : {
|
| 22 |
+
"Ios18.greaterEqual" : 1,
|
| 23 |
+
"Ios18.add" : 1,
|
| 24 |
+
"Ios18.cast" : 3,
|
| 25 |
+
"Select" : 1,
|
| 26 |
+
"Ios18.gather" : 1,
|
| 27 |
+
"Ios18.expandDims" : 2
|
| 28 |
+
},
|
| 29 |
+
"computePrecision" : "Mixed (Float16, Int16, Int32)",
|
| 30 |
+
"isUpdatable" : "0",
|
| 31 |
+
"stateSchema" : [
|
| 32 |
+
|
| 33 |
+
],
|
| 34 |
+
"availability" : {
|
| 35 |
+
"macOS" : "15.0",
|
| 36 |
+
"tvOS" : "18.0",
|
| 37 |
+
"visionOS" : "2.0",
|
| 38 |
+
"watchOS" : "11.0",
|
| 39 |
+
"iOS" : "18.0",
|
| 40 |
+
"macCatalyst" : "18.0"
|
| 41 |
+
},
|
| 42 |
+
"modelType" : {
|
| 43 |
+
"name" : "MLModelType_mlProgram"
|
| 44 |
+
},
|
| 45 |
+
"userDefinedMetadata" : {
|
| 46 |
+
"com.github.apple.coremltools.conversion_date" : "2026-03-22",
|
| 47 |
+
"com.github.apple.coremltools.source" : "torch==2.10.0",
|
| 48 |
+
"com.github.apple.coremltools.version" : "9.0",
|
| 49 |
+
"com.github.apple.coremltools.source_dialect" : "TorchScript"
|
| 50 |
+
},
|
| 51 |
+
"inputSchema" : [
|
| 52 |
+
{
|
| 53 |
+
"hasShapeFlexibility" : "0",
|
| 54 |
+
"isOptional" : "0",
|
| 55 |
+
"dataType" : "Int32",
|
| 56 |
+
"formattedType" : "MultiArray (Int32 1)",
|
| 57 |
+
"shortDescription" : "",
|
| 58 |
+
"shape" : "[1]",
|
| 59 |
+
"name" : "input_ids",
|
| 60 |
+
"type" : "MultiArray"
|
| 61 |
+
}
|
| 62 |
+
],
|
| 63 |
+
"generatedClassName" : "MultiCodeEmbedder",
|
| 64 |
+
"method" : "predict"
|
| 65 |
+
}
|
| 66 |
+
]
|
MultiCodeEmbedder.mlmodelc/model.mil
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
program(1.3)
|
| 2 |
+
[buildInfo = dict<string, string>({{"coremlc-component-MIL", "3510.2.1"}, {"coremlc-version", "3500.32.1"}, {"coremltools-component-torch", "2.10.0"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})]
|
| 3 |
+
{
|
| 4 |
+
func main<ios18>(tensor<int32, [1]> input_ids) {
|
| 5 |
+
int32 embed_batch_dims_0 = const()[name = string("embed_batch_dims_0"), val = int32(0)];
|
| 6 |
+
bool embed_validate_indices_0 = const()[name = string("embed_validate_indices_0"), val = bool(false)];
|
| 7 |
+
tensor<fp16, [30720, 1024]> embedding_weight_to_fp16 = const()[name = string("embedding_weight_to_fp16"), val = tensor<fp16, [30720, 1024]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
|
| 8 |
+
string input_ids_to_int16_dtype_0 = const()[name = string("input_ids_to_int16_dtype_0"), val = string("int16")];
|
| 9 |
+
string cast_2_dtype_0 = const()[name = string("cast_2_dtype_0"), val = string("int32")];
|
| 10 |
+
int32 greater_equal_0_y_0 = const()[name = string("greater_equal_0_y_0"), val = int32(0)];
|
| 11 |
+
tensor<int16, [1]> input_ids_to_int16 = cast(dtype = input_ids_to_int16_dtype_0, x = input_ids)[name = string("cast_5")];
|
| 12 |
+
tensor<int32, [1]> cast_2 = cast(dtype = cast_2_dtype_0, x = input_ids_to_int16)[name = string("cast_4")];
|
| 13 |
+
tensor<bool, [1]> greater_equal_0 = greater_equal(x = cast_2, y = greater_equal_0_y_0)[name = string("greater_equal_0")];
|
| 14 |
+
int32 slice_by_index_0 = const()[name = string("slice_by_index_0"), val = int32(30720)];
|
| 15 |
+
tensor<int32, [1]> add_0 = add(x = cast_2, y = slice_by_index_0)[name = string("add_0")];
|
| 16 |
+
tensor<int32, [1]> select_0 = select(a = cast_2, b = add_0, cond = greater_equal_0)[name = string("select_0")];
|
| 17 |
+
int32 embed_cast_fp16_cast_uint16_axis_0 = const()[name = string("embed_cast_fp16_cast_uint16_axis_0"), val = int32(0)];
|
| 18 |
+
string select_0_to_int16_dtype_0 = const()[name = string("select_0_to_int16_dtype_0"), val = string("int16")];
|
| 19 |
+
tensor<int16, [1]> select_0_to_int16 = cast(dtype = select_0_to_int16_dtype_0, x = select_0)[name = string("cast_3")];
|
| 20 |
+
tensor<fp16, [1, 1024]> embed_cast_fp16_cast_uint16_cast_uint16 = gather(axis = embed_cast_fp16_cast_uint16_axis_0, batch_dims = embed_batch_dims_0, indices = select_0_to_int16, validate_indices = embed_validate_indices_0, x = embedding_weight_to_fp16)[name = string("embed_cast_fp16_cast_uint16_cast_uint16")];
|
| 21 |
+
tensor<int32, [1]> var_8_axes_0 = const()[name = string("op_8_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 22 |
+
tensor<fp16, [1, 1024, 1]> var_8_cast_fp16 = expand_dims(axes = var_8_axes_0, x = embed_cast_fp16_cast_uint16_cast_uint16)[name = string("op_8_cast_fp16")];
|
| 23 |
+
tensor<int32, [1]> var_10_axes_0 = const()[name = string("op_10_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 24 |
+
tensor<fp16, [1, 1024, 1, 1]> input_embeds = expand_dims(axes = var_10_axes_0, x = var_8_cast_fp16)[name = string("op_10_cast_fp16")];
|
| 25 |
+
} -> (input_embeds);
|
| 26 |
+
}
|
MultiCodeEmbedder.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9215324f237e6e41298eaca272ea0d6f0b6c10b36ee00fd1009e03f2b193485c
|
| 3 |
+
size 62914688
|
MultiCodeEmbedder.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:51f61f175c7d56995c611824f357bc3dc048867cff6ff641edb3169138ced5cf
|
| 3 |
+
size 3351
|
MultiCodeEmbedder.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9215324f237e6e41298eaca272ea0d6f0b6c10b36ee00fd1009e03f2b193485c
|
| 3 |
+
size 62914688
|
MultiCodeEmbedder.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"67DB8580-5E2C-4286-A794-EFDDCDD4532B": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"A6587045-5979-4B84-80EA-9CE474AEB084": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Specification",
|
| 13 |
+
"name": "model.mlmodel",
|
| 14 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "A6587045-5979-4B84-80EA-9CE474AEB084"
|
| 18 |
+
}
|
SpeechDecoder.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:41f590264fbf717d778158ada892a5281aeee098a462a1b1c8f12094120d990b
|
| 3 |
+
size 243
|
SpeechDecoder.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:94b87f05a28e16baa914ea5f4cbd33078d806c696191f9f4c237758612f458ea
|
| 3 |
+
size 377
|
SpeechDecoder.mlmodelc/metadata.json
ADDED
|
@@ -0,0 +1,85 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"metadataOutputVersion" : "3.0",
|
| 4 |
+
"storagePrecision" : "Mixed (Float16, Palettized (8 bits), UInt8)",
|
| 5 |
+
"outputSchema" : [
|
| 6 |
+
{
|
| 7 |
+
"hasShapeFlexibility" : "0",
|
| 8 |
+
"isOptional" : "0",
|
| 9 |
+
"dataType" : "Float16",
|
| 10 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 240000)",
|
| 11 |
+
"shortDescription" : "",
|
| 12 |
+
"shape" : "[1, 1, 240000]",
|
| 13 |
+
"name" : "audio",
|
| 14 |
+
"type" : "MultiArray"
|
| 15 |
+
}
|
| 16 |
+
],
|
| 17 |
+
"modelParameters" : [
|
| 18 |
+
|
| 19 |
+
],
|
| 20 |
+
"specificationVersion" : 9,
|
| 21 |
+
"mlProgramOperationTypeHistogram" : {
|
| 22 |
+
"Ios18.softmax" : 8,
|
| 23 |
+
"Ios18.mul" : 174,
|
| 24 |
+
"Ios18.matmul" : 16,
|
| 25 |
+
"Ios18.rsqrt" : 17,
|
| 26 |
+
"Ios16.reduceMean" : 17,
|
| 27 |
+
"Ios18.sin" : 29,
|
| 28 |
+
"Ios18.greaterEqual" : 2,
|
| 29 |
+
"Select" : 2,
|
| 30 |
+
"Ios18.gather" : 16,
|
| 31 |
+
"Ios18.add" : 117,
|
| 32 |
+
"Ios18.layerNorm" : 2,
|
| 33 |
+
"Ios18.reshape" : 32,
|
| 34 |
+
"Pad" : 17,
|
| 35 |
+
"Ios18.constexprLutToDense" : 134,
|
| 36 |
+
"Ios18.linear" : 62,
|
| 37 |
+
"Ios18.conv" : 31,
|
| 38 |
+
"Ios18.concat" : 16,
|
| 39 |
+
"Ios18.transpose" : 40,
|
| 40 |
+
"Ios18.pow" : 46,
|
| 41 |
+
"Ios18.cast" : 19,
|
| 42 |
+
"Ios18.silu" : 8,
|
| 43 |
+
"Ios18.gelu" : 2,
|
| 44 |
+
"Ios18.clip" : 1,
|
| 45 |
+
"Ios18.convTranspose" : 6,
|
| 46 |
+
"Ios18.sliceByIndex" : 54
|
| 47 |
+
},
|
| 48 |
+
"computePrecision" : "Mixed (Float16, Float32, Int16, Int32, UInt16)",
|
| 49 |
+
"isUpdatable" : "0",
|
| 50 |
+
"stateSchema" : [
|
| 51 |
+
|
| 52 |
+
],
|
| 53 |
+
"availability" : {
|
| 54 |
+
"macOS" : "15.0",
|
| 55 |
+
"tvOS" : "18.0",
|
| 56 |
+
"visionOS" : "2.0",
|
| 57 |
+
"watchOS" : "11.0",
|
| 58 |
+
"iOS" : "18.0",
|
| 59 |
+
"macCatalyst" : "18.0"
|
| 60 |
+
},
|
| 61 |
+
"modelType" : {
|
| 62 |
+
"name" : "MLModelType_mlProgram"
|
| 63 |
+
},
|
| 64 |
+
"userDefinedMetadata" : {
|
| 65 |
+
"com.github.apple.coremltools.conversion_date" : "2026-03-22",
|
| 66 |
+
"com.github.apple.coremltools.source" : "torch==2.7.0",
|
| 67 |
+
"com.github.apple.coremltools.version" : "9.0",
|
| 68 |
+
"com.github.apple.coremltools.source_dialect" : "TorchScript"
|
| 69 |
+
},
|
| 70 |
+
"inputSchema" : [
|
| 71 |
+
{
|
| 72 |
+
"hasShapeFlexibility" : "0",
|
| 73 |
+
"isOptional" : "0",
|
| 74 |
+
"dataType" : "Int32",
|
| 75 |
+
"formattedType" : "MultiArray (Int32 1 × 16 × 125)",
|
| 76 |
+
"shortDescription" : "",
|
| 77 |
+
"shape" : "[1, 16, 125]",
|
| 78 |
+
"name" : "audio_codes",
|
| 79 |
+
"type" : "MultiArray"
|
| 80 |
+
}
|
| 81 |
+
],
|
| 82 |
+
"generatedClassName" : "SpeechDecoder",
|
| 83 |
+
"method" : "predict"
|
| 84 |
+
}
|
| 85 |
+
]
|
SpeechDecoder.mlmodelc/model.mil
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
SpeechDecoder.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9238b00d2630825fc9d176bd70e6ef8248c3732241d1af850ee2b623fd2dc715
|
| 3 |
+
size 114217664
|
SpeechDecoder.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:279b466bf61d67bddca2a3877756fd6ea8ff50c72c8ac58f2833af7e5e3d9d11
|
| 3 |
+
size 309010
|
SpeechDecoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9238b00d2630825fc9d176bd70e6ef8248c3732241d1af850ee2b623fd2dc715
|
| 3 |
+
size 114217664
|
SpeechDecoder.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"1249F3B9-82CA-4FAE-BE1D-C2CE2D4C3EC1": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"43B6139C-2FC6-4B7B-8324-95A017016304": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Specification",
|
| 13 |
+
"name": "model.mlmodel",
|
| 14 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "43B6139C-2FC6-4B7B-8324-95A017016304"
|
| 18 |
+
}
|
TextProjector.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b236e586740db8f885ed42d47f7a5671f38d0df9ab0c3933e3b92c084c5a5e18
|
| 3 |
+
size 243
|
TextProjector.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:de79900c00c95c8497ceeab0f0cdcfd975c5db6eb5ad680e874c7dd1f0a6d717
|
| 3 |
+
size 381
|
TextProjector.mlmodelc/metadata.json
ADDED
|
@@ -0,0 +1,68 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"metadataOutputVersion" : "3.0",
|
| 4 |
+
"storagePrecision" : "Float16",
|
| 5 |
+
"outputSchema" : [
|
| 6 |
+
{
|
| 7 |
+
"hasShapeFlexibility" : "0",
|
| 8 |
+
"isOptional" : "0",
|
| 9 |
+
"dataType" : "Float16",
|
| 10 |
+
"formattedType" : "MultiArray (Float16 1 × 1024 × 1 × 1)",
|
| 11 |
+
"shortDescription" : "",
|
| 12 |
+
"shape" : "[1, 1024, 1, 1]",
|
| 13 |
+
"name" : "input_embeds",
|
| 14 |
+
"type" : "MultiArray"
|
| 15 |
+
}
|
| 16 |
+
],
|
| 17 |
+
"modelParameters" : [
|
| 18 |
+
|
| 19 |
+
],
|
| 20 |
+
"specificationVersion" : 9,
|
| 21 |
+
"mlProgramOperationTypeHistogram" : {
|
| 22 |
+
"Select" : 1,
|
| 23 |
+
"Ios18.squeeze" : 1,
|
| 24 |
+
"Ios18.gather" : 1,
|
| 25 |
+
"Ios18.linear" : 2,
|
| 26 |
+
"Ios18.expandDims" : 3,
|
| 27 |
+
"Ios18.silu" : 1,
|
| 28 |
+
"Ios18.greaterEqual" : 1,
|
| 29 |
+
"Ios18.add" : 1
|
| 30 |
+
},
|
| 31 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
| 32 |
+
"isUpdatable" : "0",
|
| 33 |
+
"stateSchema" : [
|
| 34 |
+
|
| 35 |
+
],
|
| 36 |
+
"availability" : {
|
| 37 |
+
"macOS" : "15.0",
|
| 38 |
+
"tvOS" : "18.0",
|
| 39 |
+
"visionOS" : "2.0",
|
| 40 |
+
"watchOS" : "11.0",
|
| 41 |
+
"iOS" : "18.0",
|
| 42 |
+
"macCatalyst" : "18.0"
|
| 43 |
+
},
|
| 44 |
+
"modelType" : {
|
| 45 |
+
"name" : "MLModelType_mlProgram"
|
| 46 |
+
},
|
| 47 |
+
"userDefinedMetadata" : {
|
| 48 |
+
"com.github.apple.coremltools.conversion_date" : "2026-03-22",
|
| 49 |
+
"com.github.apple.coremltools.source" : "torch==2.10.0",
|
| 50 |
+
"com.github.apple.coremltools.version" : "9.0",
|
| 51 |
+
"com.github.apple.coremltools.source_dialect" : "TorchScript"
|
| 52 |
+
},
|
| 53 |
+
"inputSchema" : [
|
| 54 |
+
{
|
| 55 |
+
"hasShapeFlexibility" : "0",
|
| 56 |
+
"isOptional" : "0",
|
| 57 |
+
"dataType" : "Int32",
|
| 58 |
+
"formattedType" : "MultiArray (Int32 1)",
|
| 59 |
+
"shortDescription" : "",
|
| 60 |
+
"shape" : "[1]",
|
| 61 |
+
"name" : "input_ids",
|
| 62 |
+
"type" : "MultiArray"
|
| 63 |
+
}
|
| 64 |
+
],
|
| 65 |
+
"generatedClassName" : "TextProjector",
|
| 66 |
+
"method" : "predict"
|
| 67 |
+
}
|
| 68 |
+
]
|
TextProjector.mlmodelc/model.mil
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
program(1.3)
|
| 2 |
+
[buildInfo = dict<string, string>({{"coremlc-component-MIL", "3510.2.1"}, {"coremlc-version", "3500.32.1"}, {"coremltools-component-torch", "2.10.0"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})]
|
| 3 |
+
{
|
| 4 |
+
func main<ios18>(tensor<int32, [1]> input_ids) {
|
| 5 |
+
int32 embed_batch_dims_0 = const()[name = string("embed_batch_dims_0"), val = int32(0)];
|
| 6 |
+
bool embed_validate_indices_0 = const()[name = string("embed_validate_indices_0"), val = bool(false)];
|
| 7 |
+
tensor<fp16, [151936, 2048]> text_embedding_weight_to_fp16 = const()[name = string("text_embedding_weight_to_fp16"), val = tensor<fp16, [151936, 2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
|
| 8 |
+
int32 greater_equal_0_y_0 = const()[name = string("greater_equal_0_y_0"), val = int32(0)];
|
| 9 |
+
tensor<bool, [1]> greater_equal_0 = greater_equal(x = input_ids, y = greater_equal_0_y_0)[name = string("greater_equal_0")];
|
| 10 |
+
int32 slice_by_index_0 = const()[name = string("slice_by_index_0"), val = int32(151936)];
|
| 11 |
+
tensor<int32, [1]> add_0 = add(x = input_ids, y = slice_by_index_0)[name = string("add_0")];
|
| 12 |
+
tensor<int32, [1]> select_0 = select(a = input_ids, b = add_0, cond = greater_equal_0)[name = string("select_0")];
|
| 13 |
+
int32 embed_cast_fp16_axis_0 = const()[name = string("embed_cast_fp16_axis_0"), val = int32(0)];
|
| 14 |
+
tensor<fp16, [1, 2048]> embed_cast_fp16 = gather(axis = embed_cast_fp16_axis_0, batch_dims = embed_batch_dims_0, indices = select_0, validate_indices = embed_validate_indices_0, x = text_embedding_weight_to_fp16)[name = string("embed_cast_fp16")];
|
| 15 |
+
tensor<int32, [1]> input_1_axes_0 = const()[name = string("input_1_axes_0"), val = tensor<int32, [1]>([0])];
|
| 16 |
+
tensor<fp16, [1, 1, 2048]> input_1_cast_fp16 = expand_dims(axes = input_1_axes_0, x = embed_cast_fp16)[name = string("input_1_cast_fp16")];
|
| 17 |
+
tensor<fp16, [2048, 2048]> text_projection_linear_fc1_weight_to_fp16 = const()[name = string("text_projection_linear_fc1_weight_to_fp16"), val = tensor<fp16, [2048, 2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(622329984)))];
|
| 18 |
+
tensor<fp16, [2048]> text_projection_linear_fc1_bias_to_fp16 = const()[name = string("text_projection_linear_fc1_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(630718656)))];
|
| 19 |
+
tensor<fp16, [1, 1, 2048]> linear_0_cast_fp16 = linear(bias = text_projection_linear_fc1_bias_to_fp16, weight = text_projection_linear_fc1_weight_to_fp16, x = input_1_cast_fp16)[name = string("linear_0_cast_fp16")];
|
| 20 |
+
tensor<fp16, [1, 1, 2048]> input_cast_fp16 = silu(x = linear_0_cast_fp16)[name = string("input_cast_fp16")];
|
| 21 |
+
tensor<fp16, [1024, 2048]> text_projection_linear_fc2_weight_to_fp16 = const()[name = string("text_projection_linear_fc2_weight_to_fp16"), val = tensor<fp16, [1024, 2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(630722816)))];
|
| 22 |
+
tensor<fp16, [1024]> text_projection_linear_fc2_bias_to_fp16 = const()[name = string("text_projection_linear_fc2_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(634917184)))];
|
| 23 |
+
tensor<fp16, [1, 1, 1024]> linear_1_cast_fp16 = linear(bias = text_projection_linear_fc2_bias_to_fp16, weight = text_projection_linear_fc2_weight_to_fp16, x = input_cast_fp16)[name = string("linear_1_cast_fp16")];
|
| 24 |
+
tensor<int32, [1]> var_20_axes_0 = const()[name = string("op_20_axes_0"), val = tensor<int32, [1]>([1])];
|
| 25 |
+
tensor<fp16, [1, 1024]> var_20_cast_fp16 = squeeze(axes = var_20_axes_0, x = linear_1_cast_fp16)[name = string("op_20_cast_fp16")];
|
| 26 |
+
tensor<int32, [1]> var_22_axes_0 = const()[name = string("op_22_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 27 |
+
tensor<fp16, [1, 1024, 1]> var_22_cast_fp16 = expand_dims(axes = var_22_axes_0, x = var_20_cast_fp16)[name = string("op_22_cast_fp16")];
|
| 28 |
+
tensor<int32, [1]> var_24_axes_0 = const()[name = string("op_24_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 29 |
+
tensor<fp16, [1, 1024, 1, 1]> input_embeds = expand_dims(axes = var_24_axes_0, x = var_22_cast_fp16)[name = string("op_24_cast_fp16")];
|
| 30 |
+
} -> (input_embeds);
|
| 31 |
+
}
|
TextProjector.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:47aca0de6e40c2f27011bbb728eb7840333dd19873707f5aab75c32460aae610
|
| 3 |
+
size 634919296
|
TextProjector.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:03655ac47f8e1f7bea790679631c9f4d1b9c92cdb84e3a06319e3cf10b201067
|
| 3 |
+
size 4474
|
TextProjector.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:47aca0de6e40c2f27011bbb728eb7840333dd19873707f5aab75c32460aae610
|
| 3 |
+
size 634919296
|
TextProjector.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"202CCF22-4EC8-4090-87BA-E42DB34E6F42": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"D5A322E4-A189-4F13-B0D0-35C114E36DD0": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Specification",
|
| 13 |
+
"name": "model.mlmodel",
|
| 14 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "D5A322E4-A189-4F13-B0D0-35C114E36DD0"
|
| 18 |
+
}
|