Upload 135 files
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- .gitattributes +5 -0
- v2/_shared_mimi/.DS_Store +0 -0
- v2/_shared_mimi/mimi_decoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- v2/_shared_mimi/mimi_decoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- v2/_shared_mimi/mimi_decoder.mlpackage/Manifest.json +18 -0
- v2/english/.DS_Store +0 -0
- v2/english/cond_step.mlmodelc/analytics/coremldata.bin +3 -0
- v2/english/cond_step.mlmodelc/coremldata.bin +3 -0
- v2/english/cond_step.mlmodelc/metadata.json +317 -0
- v2/english/cond_step.mlmodelc/model.mil +0 -0
- v2/english/cond_step.mlmodelc/weights/weight.bin +3 -0
- v2/english/cond_step.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- v2/english/cond_step.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- v2/english/cond_step.mlpackage/Manifest.json +18 -0
- v2/english/constants/bos_emb.npy +3 -0
- v2/english/constants/emb_mean.npy +3 -0
- v2/english/constants/emb_std.npy +3 -0
- v2/english/constants/mimi_init_state.npz +3 -0
- v2/english/constants/quantizer_weight.npy +3 -0
- v2/english/constants/text_embed_table.npy +3 -0
- v2/english/constants_bin/alba.safetensors +3 -0
- v2/english/constants_bin/anna.safetensors +3 -0
- v2/english/constants_bin/azelma.safetensors +3 -0
- v2/english/constants_bin/bill_boerst.safetensors +3 -0
- v2/english/constants_bin/bos_emb.bin +3 -0
- v2/english/constants_bin/caro_davy.safetensors +3 -0
- v2/english/constants_bin/charles.safetensors +3 -0
- v2/english/constants_bin/cosette.safetensors +3 -0
- v2/english/constants_bin/eponine.safetensors +3 -0
- v2/english/constants_bin/eve.safetensors +3 -0
- v2/english/constants_bin/fantine.safetensors +3 -0
- v2/english/constants_bin/george.safetensors +3 -0
- v2/english/constants_bin/jane.safetensors +3 -0
- v2/english/constants_bin/javert.safetensors +3 -0
- v2/english/constants_bin/jean.safetensors +3 -0
- v2/english/constants_bin/marius.safetensors +3 -0
- v2/english/constants_bin/mary.safetensors +3 -0
- v2/english/constants_bin/michael.safetensors +3 -0
- v2/english/constants_bin/paul.safetensors +3 -0
- v2/english/constants_bin/peter_yearsley.safetensors +3 -0
- v2/english/constants_bin/stuart_bell.safetensors +3 -0
- v2/english/constants_bin/text_embed_table.bin +3 -0
- v2/english/constants_bin/tokenizer.model +3 -0
- v2/english/constants_bin/vera.safetensors +3 -0
- v2/english/flow_decoder.mlmodelc/analytics/coremldata.bin +3 -0
- v2/english/flow_decoder.mlmodelc/coremldata.bin +3 -0
- v2/english/flow_decoder.mlmodelc/metadata.json +104 -0
- v2/english/flow_decoder.mlmodelc/model.mil +312 -0
- v2/english/flow_decoder.mlmodelc/weights/weight.bin +3 -0
- v2/english/flow_decoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
.gitattributes
CHANGED
|
@@ -45,3 +45,8 @@ v2/german/verify.wav filter=lfs diff=lfs merge=lfs -text
|
|
| 45 |
v2/italian_24l/verify.wav filter=lfs diff=lfs merge=lfs -text
|
| 46 |
v2/italian/upstream.wav filter=lfs diff=lfs merge=lfs -text
|
| 47 |
v2/italian/verify.wav filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 45 |
v2/italian_24l/verify.wav filter=lfs diff=lfs merge=lfs -text
|
| 46 |
v2/italian/upstream.wav filter=lfs diff=lfs merge=lfs -text
|
| 47 |
v2/italian/verify.wav filter=lfs diff=lfs merge=lfs -text
|
| 48 |
+
v2/english/upstream_long.wav filter=lfs diff=lfs merge=lfs -text
|
| 49 |
+
v2/english/verify_long_fixed.wav filter=lfs diff=lfs merge=lfs -text
|
| 50 |
+
v2/english/verify_long.wav filter=lfs diff=lfs merge=lfs -text
|
| 51 |
+
v2/english/verify.wav filter=lfs diff=lfs merge=lfs -text
|
| 52 |
+
v2/french_24l/verify.wav filter=lfs diff=lfs merge=lfs -text
|
v2/_shared_mimi/.DS_Store
ADDED
|
Binary file (6.15 kB). View file
|
|
|
v2/_shared_mimi/mimi_decoder.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8697d7a8b97c7ccc1ed77047a1de335d0d6648bb0dd9b8f3a941d8711e73f2c1
|
| 3 |
+
size 93270
|
v2/_shared_mimi/mimi_decoder.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:c93d182f9ad8c1042ec8302093aabb4e310d76616932dd12720e03b5d07f7c64
|
| 3 |
+
size 41768256
|
v2/_shared_mimi/mimi_decoder.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"5866D77B-1BEE-4A04-B630-3AED690E11D8": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Specification",
|
| 7 |
+
"name": "model.mlmodel",
|
| 8 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 9 |
+
},
|
| 10 |
+
"AD027508-2AB2-47F8-8219-42DAA70CD43E": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Weights",
|
| 13 |
+
"name": "weights",
|
| 14 |
+
"path": "com.apple.CoreML/weights"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "5866D77B-1BEE-4A04-B630-3AED690E11D8"
|
| 18 |
+
}
|
v2/english/.DS_Store
ADDED
|
Binary file (10.2 kB). View file
|
|
|
v2/english/cond_step.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d5037f9ae66b1190ed94c2751a06b9e8cbee2c77dc1bd78dc4d5aa28852a31f6
|
| 3 |
+
size 243
|
v2/english/cond_step.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4373d5077fdfdd5df7ade208a864207cc867718b0725e6a82b3c125cb7407a8d
|
| 3 |
+
size 1125
|
v2/english/cond_step.mlmodelc/metadata.json
ADDED
|
@@ -0,0 +1,317 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"metadataOutputVersion" : "3.0",
|
| 4 |
+
"storagePrecision" : "Float32",
|
| 5 |
+
"outputSchema" : [
|
| 6 |
+
{
|
| 7 |
+
"hasShapeFlexibility" : "0",
|
| 8 |
+
"isOptional" : "0",
|
| 9 |
+
"dataType" : "Float32",
|
| 10 |
+
"formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)",
|
| 11 |
+
"shortDescription" : "",
|
| 12 |
+
"shape" : "[2, 1, 512, 16, 64]",
|
| 13 |
+
"name" : "new_cache_1_internal_tensor_assign_2",
|
| 14 |
+
"type" : "MultiArray"
|
| 15 |
+
},
|
| 16 |
+
{
|
| 17 |
+
"hasShapeFlexibility" : "0",
|
| 18 |
+
"isOptional" : "0",
|
| 19 |
+
"dataType" : "Float32",
|
| 20 |
+
"formattedType" : "MultiArray (Float32 1)",
|
| 21 |
+
"shortDescription" : "",
|
| 22 |
+
"shape" : "[1]",
|
| 23 |
+
"name" : "var_445",
|
| 24 |
+
"type" : "MultiArray"
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"hasShapeFlexibility" : "0",
|
| 28 |
+
"isOptional" : "0",
|
| 29 |
+
"dataType" : "Float32",
|
| 30 |
+
"formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)",
|
| 31 |
+
"shortDescription" : "",
|
| 32 |
+
"shape" : "[2, 1, 512, 16, 64]",
|
| 33 |
+
"name" : "new_cache_3_internal_tensor_assign_2",
|
| 34 |
+
"type" : "MultiArray"
|
| 35 |
+
},
|
| 36 |
+
{
|
| 37 |
+
"hasShapeFlexibility" : "0",
|
| 38 |
+
"isOptional" : "0",
|
| 39 |
+
"dataType" : "Float32",
|
| 40 |
+
"formattedType" : "MultiArray (Float32 1)",
|
| 41 |
+
"shortDescription" : "",
|
| 42 |
+
"shape" : "[1]",
|
| 43 |
+
"name" : "var_864",
|
| 44 |
+
"type" : "MultiArray"
|
| 45 |
+
},
|
| 46 |
+
{
|
| 47 |
+
"hasShapeFlexibility" : "0",
|
| 48 |
+
"isOptional" : "0",
|
| 49 |
+
"dataType" : "Float32",
|
| 50 |
+
"formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)",
|
| 51 |
+
"shortDescription" : "",
|
| 52 |
+
"shape" : "[2, 1, 512, 16, 64]",
|
| 53 |
+
"name" : "new_cache_5_internal_tensor_assign_2",
|
| 54 |
+
"type" : "MultiArray"
|
| 55 |
+
},
|
| 56 |
+
{
|
| 57 |
+
"hasShapeFlexibility" : "0",
|
| 58 |
+
"isOptional" : "0",
|
| 59 |
+
"dataType" : "Float32",
|
| 60 |
+
"formattedType" : "MultiArray (Float32 1)",
|
| 61 |
+
"shortDescription" : "",
|
| 62 |
+
"shape" : "[1]",
|
| 63 |
+
"name" : "var_1283",
|
| 64 |
+
"type" : "MultiArray"
|
| 65 |
+
},
|
| 66 |
+
{
|
| 67 |
+
"hasShapeFlexibility" : "0",
|
| 68 |
+
"isOptional" : "0",
|
| 69 |
+
"dataType" : "Float32",
|
| 70 |
+
"formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)",
|
| 71 |
+
"shortDescription" : "",
|
| 72 |
+
"shape" : "[2, 1, 512, 16, 64]",
|
| 73 |
+
"name" : "new_cache_7_internal_tensor_assign_2",
|
| 74 |
+
"type" : "MultiArray"
|
| 75 |
+
},
|
| 76 |
+
{
|
| 77 |
+
"hasShapeFlexibility" : "0",
|
| 78 |
+
"isOptional" : "0",
|
| 79 |
+
"dataType" : "Float32",
|
| 80 |
+
"formattedType" : "MultiArray (Float32 1)",
|
| 81 |
+
"shortDescription" : "",
|
| 82 |
+
"shape" : "[1]",
|
| 83 |
+
"name" : "var_1702",
|
| 84 |
+
"type" : "MultiArray"
|
| 85 |
+
},
|
| 86 |
+
{
|
| 87 |
+
"hasShapeFlexibility" : "0",
|
| 88 |
+
"isOptional" : "0",
|
| 89 |
+
"dataType" : "Float32",
|
| 90 |
+
"formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)",
|
| 91 |
+
"shortDescription" : "",
|
| 92 |
+
"shape" : "[2, 1, 512, 16, 64]",
|
| 93 |
+
"name" : "new_cache_9_internal_tensor_assign_2",
|
| 94 |
+
"type" : "MultiArray"
|
| 95 |
+
},
|
| 96 |
+
{
|
| 97 |
+
"hasShapeFlexibility" : "0",
|
| 98 |
+
"isOptional" : "0",
|
| 99 |
+
"dataType" : "Float32",
|
| 100 |
+
"formattedType" : "MultiArray (Float32 1)",
|
| 101 |
+
"shortDescription" : "",
|
| 102 |
+
"shape" : "[1]",
|
| 103 |
+
"name" : "var_2121",
|
| 104 |
+
"type" : "MultiArray"
|
| 105 |
+
},
|
| 106 |
+
{
|
| 107 |
+
"hasShapeFlexibility" : "0",
|
| 108 |
+
"isOptional" : "0",
|
| 109 |
+
"dataType" : "Float32",
|
| 110 |
+
"formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)",
|
| 111 |
+
"shortDescription" : "",
|
| 112 |
+
"shape" : "[2, 1, 512, 16, 64]",
|
| 113 |
+
"name" : "new_cache_internal_tensor_assign_2",
|
| 114 |
+
"type" : "MultiArray"
|
| 115 |
+
},
|
| 116 |
+
{
|
| 117 |
+
"hasShapeFlexibility" : "0",
|
| 118 |
+
"isOptional" : "0",
|
| 119 |
+
"dataType" : "Float32",
|
| 120 |
+
"formattedType" : "MultiArray (Float32 1)",
|
| 121 |
+
"shortDescription" : "",
|
| 122 |
+
"shape" : "[1]",
|
| 123 |
+
"name" : "var_2365",
|
| 124 |
+
"type" : "MultiArray"
|
| 125 |
+
}
|
| 126 |
+
],
|
| 127 |
+
"modelParameters" : [
|
| 128 |
+
|
| 129 |
+
],
|
| 130 |
+
"specificationVersion" : 8,
|
| 131 |
+
"mlProgramOperationTypeHistogram" : {
|
| 132 |
+
"Ios17.cos" : 6,
|
| 133 |
+
"Range1d" : 12,
|
| 134 |
+
"Ios17.notEqual" : 10,
|
| 135 |
+
"Ios17.floor" : 6,
|
| 136 |
+
"Ios17.reshape" : 104,
|
| 137 |
+
"Ios17.logicalAnd" : 5,
|
| 138 |
+
"Ios16.softmax" : 5,
|
| 139 |
+
"Ios17.scatter" : 12,
|
| 140 |
+
"Ios17.matmul" : 10,
|
| 141 |
+
"Ios17.sin" : 6,
|
| 142 |
+
"Ios17.transpose" : 20,
|
| 143 |
+
"Select" : 15,
|
| 144 |
+
"Ios17.expandDims" : 5,
|
| 145 |
+
"Ios17.add" : 32,
|
| 146 |
+
"Tile" : 6,
|
| 147 |
+
"Ios17.lessEqual" : 5,
|
| 148 |
+
"Ios17.sliceByIndex" : 73,
|
| 149 |
+
"Ios17.scatterAlongAxis" : 12,
|
| 150 |
+
"Ios17.logicalNot" : 5,
|
| 151 |
+
"Ios17.layerNorm" : 11,
|
| 152 |
+
"Ios17.less" : 5,
|
| 153 |
+
"Ios17.sub" : 17,
|
| 154 |
+
"Ios17.cast" : 6,
|
| 155 |
+
"Ios16.gelu" : 5,
|
| 156 |
+
"Ios17.linear" : 21,
|
| 157 |
+
"Stack" : 11,
|
| 158 |
+
"Ios17.mul" : 67
|
| 159 |
+
},
|
| 160 |
+
"computePrecision" : "Mixed (Float32, Int32)",
|
| 161 |
+
"isUpdatable" : "0",
|
| 162 |
+
"stateSchema" : [
|
| 163 |
+
|
| 164 |
+
],
|
| 165 |
+
"availability" : {
|
| 166 |
+
"macOS" : "14.0",
|
| 167 |
+
"tvOS" : "17.0",
|
| 168 |
+
"visionOS" : "1.0",
|
| 169 |
+
"watchOS" : "10.0",
|
| 170 |
+
"iOS" : "17.0",
|
| 171 |
+
"macCatalyst" : "17.0"
|
| 172 |
+
},
|
| 173 |
+
"modelType" : {
|
| 174 |
+
"name" : "MLModelType_mlProgram"
|
| 175 |
+
},
|
| 176 |
+
"userDefinedMetadata" : {
|
| 177 |
+
"com.github.apple.coremltools.conversion_date" : "2026-04-24",
|
| 178 |
+
"com.github.apple.coremltools.source" : "torch==2.9.1",
|
| 179 |
+
"com.github.apple.coremltools.version" : "9.0",
|
| 180 |
+
"com.github.apple.coremltools.source_dialect" : "TorchScript"
|
| 181 |
+
},
|
| 182 |
+
"inputSchema" : [
|
| 183 |
+
{
|
| 184 |
+
"hasShapeFlexibility" : "0",
|
| 185 |
+
"isOptional" : "0",
|
| 186 |
+
"dataType" : "Float32",
|
| 187 |
+
"formattedType" : "MultiArray (Float32 1 × 1 × 1024)",
|
| 188 |
+
"shortDescription" : "",
|
| 189 |
+
"shape" : "[1, 1, 1024]",
|
| 190 |
+
"name" : "conditioning",
|
| 191 |
+
"type" : "MultiArray"
|
| 192 |
+
},
|
| 193 |
+
{
|
| 194 |
+
"hasShapeFlexibility" : "0",
|
| 195 |
+
"isOptional" : "0",
|
| 196 |
+
"dataType" : "Float32",
|
| 197 |
+
"formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)",
|
| 198 |
+
"shortDescription" : "",
|
| 199 |
+
"shape" : "[2, 1, 512, 16, 64]",
|
| 200 |
+
"name" : "cache0",
|
| 201 |
+
"type" : "MultiArray"
|
| 202 |
+
},
|
| 203 |
+
{
|
| 204 |
+
"hasShapeFlexibility" : "0",
|
| 205 |
+
"isOptional" : "0",
|
| 206 |
+
"dataType" : "Float32",
|
| 207 |
+
"formattedType" : "MultiArray (Float32 1)",
|
| 208 |
+
"shortDescription" : "",
|
| 209 |
+
"shape" : "[1]",
|
| 210 |
+
"name" : "position0",
|
| 211 |
+
"type" : "MultiArray"
|
| 212 |
+
},
|
| 213 |
+
{
|
| 214 |
+
"hasShapeFlexibility" : "0",
|
| 215 |
+
"isOptional" : "0",
|
| 216 |
+
"dataType" : "Float32",
|
| 217 |
+
"formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)",
|
| 218 |
+
"shortDescription" : "",
|
| 219 |
+
"shape" : "[2, 1, 512, 16, 64]",
|
| 220 |
+
"name" : "cache1",
|
| 221 |
+
"type" : "MultiArray"
|
| 222 |
+
},
|
| 223 |
+
{
|
| 224 |
+
"hasShapeFlexibility" : "0",
|
| 225 |
+
"isOptional" : "0",
|
| 226 |
+
"dataType" : "Float32",
|
| 227 |
+
"formattedType" : "MultiArray (Float32 1)",
|
| 228 |
+
"shortDescription" : "",
|
| 229 |
+
"shape" : "[1]",
|
| 230 |
+
"name" : "position1",
|
| 231 |
+
"type" : "MultiArray"
|
| 232 |
+
},
|
| 233 |
+
{
|
| 234 |
+
"hasShapeFlexibility" : "0",
|
| 235 |
+
"isOptional" : "0",
|
| 236 |
+
"dataType" : "Float32",
|
| 237 |
+
"formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)",
|
| 238 |
+
"shortDescription" : "",
|
| 239 |
+
"shape" : "[2, 1, 512, 16, 64]",
|
| 240 |
+
"name" : "cache2",
|
| 241 |
+
"type" : "MultiArray"
|
| 242 |
+
},
|
| 243 |
+
{
|
| 244 |
+
"hasShapeFlexibility" : "0",
|
| 245 |
+
"isOptional" : "0",
|
| 246 |
+
"dataType" : "Float32",
|
| 247 |
+
"formattedType" : "MultiArray (Float32 1)",
|
| 248 |
+
"shortDescription" : "",
|
| 249 |
+
"shape" : "[1]",
|
| 250 |
+
"name" : "position2",
|
| 251 |
+
"type" : "MultiArray"
|
| 252 |
+
},
|
| 253 |
+
{
|
| 254 |
+
"hasShapeFlexibility" : "0",
|
| 255 |
+
"isOptional" : "0",
|
| 256 |
+
"dataType" : "Float32",
|
| 257 |
+
"formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)",
|
| 258 |
+
"shortDescription" : "",
|
| 259 |
+
"shape" : "[2, 1, 512, 16, 64]",
|
| 260 |
+
"name" : "cache3",
|
| 261 |
+
"type" : "MultiArray"
|
| 262 |
+
},
|
| 263 |
+
{
|
| 264 |
+
"hasShapeFlexibility" : "0",
|
| 265 |
+
"isOptional" : "0",
|
| 266 |
+
"dataType" : "Float32",
|
| 267 |
+
"formattedType" : "MultiArray (Float32 1)",
|
| 268 |
+
"shortDescription" : "",
|
| 269 |
+
"shape" : "[1]",
|
| 270 |
+
"name" : "position3",
|
| 271 |
+
"type" : "MultiArray"
|
| 272 |
+
},
|
| 273 |
+
{
|
| 274 |
+
"hasShapeFlexibility" : "0",
|
| 275 |
+
"isOptional" : "0",
|
| 276 |
+
"dataType" : "Float32",
|
| 277 |
+
"formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)",
|
| 278 |
+
"shortDescription" : "",
|
| 279 |
+
"shape" : "[2, 1, 512, 16, 64]",
|
| 280 |
+
"name" : "cache4",
|
| 281 |
+
"type" : "MultiArray"
|
| 282 |
+
},
|
| 283 |
+
{
|
| 284 |
+
"hasShapeFlexibility" : "0",
|
| 285 |
+
"isOptional" : "0",
|
| 286 |
+
"dataType" : "Float32",
|
| 287 |
+
"formattedType" : "MultiArray (Float32 1)",
|
| 288 |
+
"shortDescription" : "",
|
| 289 |
+
"shape" : "[1]",
|
| 290 |
+
"name" : "position4",
|
| 291 |
+
"type" : "MultiArray"
|
| 292 |
+
},
|
| 293 |
+
{
|
| 294 |
+
"hasShapeFlexibility" : "0",
|
| 295 |
+
"isOptional" : "0",
|
| 296 |
+
"dataType" : "Float32",
|
| 297 |
+
"formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)",
|
| 298 |
+
"shortDescription" : "",
|
| 299 |
+
"shape" : "[2, 1, 512, 16, 64]",
|
| 300 |
+
"name" : "cache5",
|
| 301 |
+
"type" : "MultiArray"
|
| 302 |
+
},
|
| 303 |
+
{
|
| 304 |
+
"hasShapeFlexibility" : "0",
|
| 305 |
+
"isOptional" : "0",
|
| 306 |
+
"dataType" : "Float32",
|
| 307 |
+
"formattedType" : "MultiArray (Float32 1)",
|
| 308 |
+
"shortDescription" : "",
|
| 309 |
+
"shape" : "[1]",
|
| 310 |
+
"name" : "position5",
|
| 311 |
+
"type" : "MultiArray"
|
| 312 |
+
}
|
| 313 |
+
],
|
| 314 |
+
"generatedClassName" : "cond_step",
|
| 315 |
+
"method" : "predict"
|
| 316 |
+
}
|
| 317 |
+
]
|
v2/english/cond_step.mlmodelc/model.mil
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
v2/english/cond_step.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4f49f0b615e3856b2ec647d3699220893c2be441222694d5c76647d553e69e4b
|
| 3 |
+
size 266467520
|
v2/english/cond_step.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:367c13471a712bba041b013fce8b718dc50389df9c46af5638e49a4b7ca224f2
|
| 3 |
+
size 167298
|
v2/english/cond_step.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:4f49f0b615e3856b2ec647d3699220893c2be441222694d5c76647d553e69e4b
|
| 3 |
+
size 266467520
|
v2/english/cond_step.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"7C979841-0955-417B-B3AE-B8B49287B2B4": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Specification",
|
| 7 |
+
"name": "model.mlmodel",
|
| 8 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 9 |
+
},
|
| 10 |
+
"84FA66B3-414D-46C8-86C9-F01E9FC542BE": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Weights",
|
| 13 |
+
"name": "weights",
|
| 14 |
+
"path": "com.apple.CoreML/weights"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "7C979841-0955-417B-B3AE-B8B49287B2B4"
|
| 18 |
+
}
|
v2/english/constants/bos_emb.npy
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4ae6121755d51695873cda7822b0959cba18b28543439a478b39d5c51b91e749
|
| 3 |
+
size 256
|
v2/english/constants/emb_mean.npy
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:600795312716d8776c2a3f438d83c458ab5f4ac64128cd0eb568079d8c9b8e9b
|
| 3 |
+
size 256
|
v2/english/constants/emb_std.npy
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:638c529cedbeef6c87739f56c5de4d7635a24bc43f13f6fec13dab804d3a208d
|
| 3 |
+
size 256
|
v2/english/constants/mimi_init_state.npz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:515588e10e9c14a6b009f1cfda9fcd592542b1640ad856d5fc69e50024702a39
|
| 3 |
+
size 1083108
|
v2/english/constants/quantizer_weight.npy
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3cb7b1a032c562af193adf9ffdd3366aca8f275da1f48ce0456a122a8c688f03
|
| 3 |
+
size 65664
|
v2/english/constants/text_embed_table.npy
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0747d8e12ab8798d09be0ee390827b895b6d1babb82037b28aaca03aed8b2203
|
| 3 |
+
size 16388224
|
v2/english/constants_bin/alba.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e4a323e8e771f14fd669428d83e7af6d83b4d6d0f0e282c30122f04824f17b1c
|
| 3 |
+
size 6194424
|
v2/english/constants_bin/anna.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:87e1922e9d1bfb27728ae4f1ea89bc3174e17a9f819ce0a728cfbea19517d9c7
|
| 3 |
+
size 7816440
|
v2/english/constants_bin/azelma.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3d4ede5a1dbfd3632d2b9a5738dc3cc60ce799646f5c110ede9c8162e4eb289f
|
| 3 |
+
size 7963896
|
v2/english/constants_bin/bill_boerst.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a2eb24310d563b4600573f3afdffb2c871bc6c1392beffcddbcb3c92e605483a
|
| 3 |
+
size 6735096
|
v2/english/constants_bin/bos_emb.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:52f1a157fa35fc100213e6e0fafa74703be99db8faa42de3e99feda607ac7b63
|
| 3 |
+
size 128
|
v2/english/constants_bin/caro_davy.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e5217ce9629f3473b27519dc280e80c79eb304853f5e4104ea7cd65f77d3dc5f
|
| 3 |
+
size 5260536
|
v2/english/constants_bin/charles.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0681cdd3decb917df78f7ebeb42f01f2784d7490911591d8f93af906ab3770e5
|
| 3 |
+
size 6194424
|
v2/english/constants_bin/cosette.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:014818ee1fe5ef13e11f238b17822d27021b7cfd7e264402e49087d24250fe7f
|
| 3 |
+
size 6194424
|
v2/english/constants_bin/eponine.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e4ae23d204cfc26f47a132d70051bb99331905e3d45fc2a71f5eb356bd7d1e27
|
| 3 |
+
size 6931704
|
v2/english/constants_bin/eve.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a9509d0951b2ed9ac9b1d08a75199aba88c974e3e74cf6168d251bb7ccdc01e7
|
| 3 |
+
size 6538488
|
v2/english/constants_bin/fantine.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b7581a77c09d39b43a0b619b1962042e6469e21cf13cdcce67b880bec4ff46fd
|
| 3 |
+
size 6538488
|
v2/english/constants_bin/george.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9a78363b01ee91dcfd431b500e84350ef6f8e6e0639d5a34f1441cd6129d4dc7
|
| 3 |
+
size 6243576
|
v2/english/constants_bin/jane.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d4354fc55d32045f7ca45e616ddd63c64e27487f743c098765345cb50cfbd78b
|
| 3 |
+
size 7374072
|
v2/english/constants_bin/javert.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d476bdc0b43b42192f4934ea080553851d1e5897d9eefe965e7a30b69b7d10a3
|
| 3 |
+
size 6194424
|
v2/english/constants_bin/jean.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:82a9d8ee6d371da90c9b6400629aba7d77003c3ee01e12a8635caee32aff964c
|
| 3 |
+
size 6194424
|
v2/english/constants_bin/marius.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6e561ad98e5c36b4fae278e9a8540d4b86069c46446028efd04e9d58fa25d3c9
|
| 3 |
+
size 6194424
|
v2/english/constants_bin/mary.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4e1de7cab5058ccd0d9f1a7f469674502354d7c0643f2d1a43e5c268278be0f0
|
| 3 |
+
size 6194424
|
v2/english/constants_bin/michael.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e23d5ae79a6c0e96f13bf60267d88a99c893bda9e6d559de4e78135060935bd7
|
| 3 |
+
size 7275768
|
v2/english/constants_bin/paul.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8c227d54297a5be31ecae4fe9067c4f7fba49606c241cfe70279eb9cb67b9e46
|
| 3 |
+
size 6980856
|
v2/english/constants_bin/peter_yearsley.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:837af0682b00e268bf034ccbf0400125369bbd0d4e2d0c3a3ddea1487acbc543
|
| 3 |
+
size 3736816
|
v2/english/constants_bin/stuart_bell.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7c6f0f5bc4806cc5e852e6af96ac386a9679362e35bbf8c3dc9b8fefa04ed055
|
| 3 |
+
size 5260536
|
v2/english/constants_bin/text_embed_table.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2512f9edd82ab0c68bb42a5fcd329a371eaf63ee6fb5275dfb1bc9536592bb0e
|
| 3 |
+
size 16388096
|
v2/english/constants_bin/tokenizer.model
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d461765ae179566678c93091c5fa6f2984c31bbe990bf1aa62d92c64d91bc3f6
|
| 3 |
+
size 59339
|
v2/english/constants_bin/vera.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c1409379aa12ab8a88b97b83270896d721efc1b4dbad8df7a2094b2b69cbfab6
|
| 3 |
+
size 6735096
|
v2/english/flow_decoder.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:18b5ee269d916881452291a3fbc059a39a1ff5c8827e0c2a3806262a1f78caa1
|
| 3 |
+
size 243
|
v2/english/flow_decoder.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:86653dc204e0133d89738c1b5873b927f1fab36e2f1694f7e4b788cf76cd5e18
|
| 3 |
+
size 436
|
v2/english/flow_decoder.mlmodelc/metadata.json
ADDED
|
@@ -0,0 +1,104 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"metadataOutputVersion" : "3.0",
|
| 4 |
+
"storagePrecision" : "Float32",
|
| 5 |
+
"outputSchema" : [
|
| 6 |
+
{
|
| 7 |
+
"hasShapeFlexibility" : "0",
|
| 8 |
+
"isOptional" : "0",
|
| 9 |
+
"dataType" : "Float32",
|
| 10 |
+
"formattedType" : "MultiArray (Float32 1 × 32)",
|
| 11 |
+
"shortDescription" : "",
|
| 12 |
+
"shape" : "[1, 32]",
|
| 13 |
+
"name" : "var_358",
|
| 14 |
+
"type" : "MultiArray"
|
| 15 |
+
}
|
| 16 |
+
],
|
| 17 |
+
"modelParameters" : [
|
| 18 |
+
|
| 19 |
+
],
|
| 20 |
+
"specificationVersion" : 8,
|
| 21 |
+
"mlProgramOperationTypeHistogram" : {
|
| 22 |
+
"Ios17.mul" : 28,
|
| 23 |
+
"Ios17.sqrt" : 7,
|
| 24 |
+
"Ios17.linear" : 26,
|
| 25 |
+
"Ios17.cos" : 2,
|
| 26 |
+
"Ios17.rsqrt" : 2,
|
| 27 |
+
"Ios17.sub" : 9,
|
| 28 |
+
"Ios17.square" : 9,
|
| 29 |
+
"Ios17.sin" : 2,
|
| 30 |
+
"Ios17.concat" : 2,
|
| 31 |
+
"Ios17.add" : 37,
|
| 32 |
+
"Ios16.reduceMean" : 18,
|
| 33 |
+
"Ios17.realDiv" : 7,
|
| 34 |
+
"Split" : 7,
|
| 35 |
+
"Ios16.silu" : 9
|
| 36 |
+
},
|
| 37 |
+
"computePrecision" : "Mixed (Float32, Int32)",
|
| 38 |
+
"isUpdatable" : "0",
|
| 39 |
+
"stateSchema" : [
|
| 40 |
+
|
| 41 |
+
],
|
| 42 |
+
"availability" : {
|
| 43 |
+
"macOS" : "14.0",
|
| 44 |
+
"tvOS" : "17.0",
|
| 45 |
+
"visionOS" : "1.0",
|
| 46 |
+
"watchOS" : "10.0",
|
| 47 |
+
"iOS" : "17.0",
|
| 48 |
+
"macCatalyst" : "17.0"
|
| 49 |
+
},
|
| 50 |
+
"modelType" : {
|
| 51 |
+
"name" : "MLModelType_mlProgram"
|
| 52 |
+
},
|
| 53 |
+
"userDefinedMetadata" : {
|
| 54 |
+
"com.github.apple.coremltools.conversion_date" : "2026-04-24",
|
| 55 |
+
"com.github.apple.coremltools.source" : "torch==2.9.1",
|
| 56 |
+
"com.github.apple.coremltools.version" : "9.0",
|
| 57 |
+
"com.github.apple.coremltools.source_dialect" : "TorchScript"
|
| 58 |
+
},
|
| 59 |
+
"inputSchema" : [
|
| 60 |
+
{
|
| 61 |
+
"hasShapeFlexibility" : "0",
|
| 62 |
+
"isOptional" : "0",
|
| 63 |
+
"dataType" : "Float32",
|
| 64 |
+
"formattedType" : "MultiArray (Float32 1 × 1024)",
|
| 65 |
+
"shortDescription" : "",
|
| 66 |
+
"shape" : "[1, 1024]",
|
| 67 |
+
"name" : "transformer_out",
|
| 68 |
+
"type" : "MultiArray"
|
| 69 |
+
},
|
| 70 |
+
{
|
| 71 |
+
"hasShapeFlexibility" : "0",
|
| 72 |
+
"isOptional" : "0",
|
| 73 |
+
"dataType" : "Float32",
|
| 74 |
+
"formattedType" : "MultiArray (Float32 1 × 32)",
|
| 75 |
+
"shortDescription" : "",
|
| 76 |
+
"shape" : "[1, 32]",
|
| 77 |
+
"name" : "latent",
|
| 78 |
+
"type" : "MultiArray"
|
| 79 |
+
},
|
| 80 |
+
{
|
| 81 |
+
"hasShapeFlexibility" : "0",
|
| 82 |
+
"isOptional" : "0",
|
| 83 |
+
"dataType" : "Float32",
|
| 84 |
+
"formattedType" : "MultiArray (Float32 1 × 1)",
|
| 85 |
+
"shortDescription" : "",
|
| 86 |
+
"shape" : "[1, 1]",
|
| 87 |
+
"name" : "s",
|
| 88 |
+
"type" : "MultiArray"
|
| 89 |
+
},
|
| 90 |
+
{
|
| 91 |
+
"hasShapeFlexibility" : "0",
|
| 92 |
+
"isOptional" : "0",
|
| 93 |
+
"dataType" : "Float32",
|
| 94 |
+
"formattedType" : "MultiArray (Float32 1 × 1)",
|
| 95 |
+
"shortDescription" : "",
|
| 96 |
+
"shape" : "[1, 1]",
|
| 97 |
+
"name" : "t",
|
| 98 |
+
"type" : "MultiArray"
|
| 99 |
+
}
|
| 100 |
+
],
|
| 101 |
+
"generatedClassName" : "flow_decoder",
|
| 102 |
+
"method" : "predict"
|
| 103 |
+
}
|
| 104 |
+
]
|
v2/english/flow_decoder.mlmodelc/model.mil
ADDED
|
@@ -0,0 +1,312 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
program(1.0)
|
| 2 |
+
[buildInfo = dict<tensor<string, []>, tensor<string, []>>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}, {"coremltools-component-torch", "2.9.1"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})]
|
| 3 |
+
{
|
| 4 |
+
func main<ios17>(tensor<fp32, [1, 32]> latent, tensor<fp32, [1, 1]> s, tensor<fp32, [1, 1]> t, tensor<fp32, [1, 1024]> transformer_out) {
|
| 5 |
+
tensor<fp32, [512]> flow_net_input_proj_bias = const()[name = tensor<string, []>("flow_net_input_proj_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(64)))];
|
| 6 |
+
tensor<fp32, [512, 32]> flow_net_input_proj_weight = const()[name = tensor<string, []>("flow_net_input_proj_weight"), val = tensor<fp32, [512, 32]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2176)))];
|
| 7 |
+
tensor<fp32, [512]> flow_net_time_embed_0_mlp_0_bias = const()[name = tensor<string, []>("flow_net_time_embed_0_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(67776)))];
|
| 8 |
+
tensor<fp32, [512, 256]> flow_net_time_embed_0_mlp_0_weight = const()[name = tensor<string, []>("flow_net_time_embed_0_mlp_0_weight"), val = tensor<fp32, [512, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(69888)))];
|
| 9 |
+
tensor<fp32, [512]> flow_net_time_embed_0_mlp_2_bias = const()[name = tensor<string, []>("flow_net_time_embed_0_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(594240)))];
|
| 10 |
+
tensor<fp32, [512, 512]> flow_net_time_embed_0_mlp_2_weight = const()[name = tensor<string, []>("flow_net_time_embed_0_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(596352)))];
|
| 11 |
+
tensor<fp32, [512]> flow_net_time_embed_1_mlp_0_bias = const()[name = tensor<string, []>("flow_net_time_embed_1_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1644992)))];
|
| 12 |
+
tensor<fp32, [512, 256]> flow_net_time_embed_1_mlp_0_weight = const()[name = tensor<string, []>("flow_net_time_embed_1_mlp_0_weight"), val = tensor<fp32, [512, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1647104)))];
|
| 13 |
+
tensor<fp32, [512]> flow_net_time_embed_1_mlp_2_bias = const()[name = tensor<string, []>("flow_net_time_embed_1_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2171456)))];
|
| 14 |
+
tensor<fp32, [512, 512]> flow_net_time_embed_1_mlp_2_weight = const()[name = tensor<string, []>("flow_net_time_embed_1_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2173568)))];
|
| 15 |
+
tensor<fp32, [512]> flow_net_cond_embed_bias = const()[name = tensor<string, []>("flow_net_cond_embed_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3222208)))];
|
| 16 |
+
tensor<fp32, [512, 1024]> flow_net_cond_embed_weight = const()[name = tensor<string, []>("flow_net_cond_embed_weight"), val = tensor<fp32, [512, 1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3224320)))];
|
| 17 |
+
tensor<fp32, [1536]> flow_net_res_blocks_0_adaLN_modulation_1_bias = const()[name = tensor<string, []>("flow_net_res_blocks_0_adaLN_modulation_1_bias"), val = tensor<fp32, [1536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5321536)))];
|
| 18 |
+
tensor<fp32, [1536, 512]> flow_net_res_blocks_0_adaLN_modulation_1_weight = const()[name = tensor<string, []>("flow_net_res_blocks_0_adaLN_modulation_1_weight"), val = tensor<fp32, [1536, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5327744)))];
|
| 19 |
+
tensor<fp32, [512]> flow_net_res_blocks_0_in_ln_bias = const()[name = tensor<string, []>("flow_net_res_blocks_0_in_ln_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8473536)))];
|
| 20 |
+
tensor<fp32, [512]> flow_net_res_blocks_0_in_ln_weight = const()[name = tensor<string, []>("flow_net_res_blocks_0_in_ln_weight"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8475648)))];
|
| 21 |
+
tensor<fp32, [512]> flow_net_res_blocks_0_mlp_0_bias = const()[name = tensor<string, []>("flow_net_res_blocks_0_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8477760)))];
|
| 22 |
+
tensor<fp32, [512, 512]> flow_net_res_blocks_0_mlp_0_weight = const()[name = tensor<string, []>("flow_net_res_blocks_0_mlp_0_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8479872)))];
|
| 23 |
+
tensor<fp32, [512]> flow_net_res_blocks_0_mlp_2_bias = const()[name = tensor<string, []>("flow_net_res_blocks_0_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9528512)))];
|
| 24 |
+
tensor<fp32, [512, 512]> flow_net_res_blocks_0_mlp_2_weight = const()[name = tensor<string, []>("flow_net_res_blocks_0_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9530624)))];
|
| 25 |
+
tensor<fp32, [1536]> flow_net_res_blocks_1_adaLN_modulation_1_bias = const()[name = tensor<string, []>("flow_net_res_blocks_1_adaLN_modulation_1_bias"), val = tensor<fp32, [1536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10579264)))];
|
| 26 |
+
tensor<fp32, [1536, 512]> flow_net_res_blocks_1_adaLN_modulation_1_weight = const()[name = tensor<string, []>("flow_net_res_blocks_1_adaLN_modulation_1_weight"), val = tensor<fp32, [1536, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10585472)))];
|
| 27 |
+
tensor<fp32, [512]> flow_net_res_blocks_1_in_ln_bias = const()[name = tensor<string, []>("flow_net_res_blocks_1_in_ln_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13731264)))];
|
| 28 |
+
tensor<fp32, [512]> flow_net_res_blocks_1_in_ln_weight = const()[name = tensor<string, []>("flow_net_res_blocks_1_in_ln_weight"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13733376)))];
|
| 29 |
+
tensor<fp32, [512]> flow_net_res_blocks_1_mlp_0_bias = const()[name = tensor<string, []>("flow_net_res_blocks_1_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13735488)))];
|
| 30 |
+
tensor<fp32, [512, 512]> flow_net_res_blocks_1_mlp_0_weight = const()[name = tensor<string, []>("flow_net_res_blocks_1_mlp_0_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13737600)))];
|
| 31 |
+
tensor<fp32, [512]> flow_net_res_blocks_1_mlp_2_bias = const()[name = tensor<string, []>("flow_net_res_blocks_1_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(14786240)))];
|
| 32 |
+
tensor<fp32, [512, 512]> flow_net_res_blocks_1_mlp_2_weight = const()[name = tensor<string, []>("flow_net_res_blocks_1_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(14788352)))];
|
| 33 |
+
tensor<fp32, [1536]> flow_net_res_blocks_2_adaLN_modulation_1_bias = const()[name = tensor<string, []>("flow_net_res_blocks_2_adaLN_modulation_1_bias"), val = tensor<fp32, [1536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(15836992)))];
|
| 34 |
+
tensor<fp32, [1536, 512]> flow_net_res_blocks_2_adaLN_modulation_1_weight = const()[name = tensor<string, []>("flow_net_res_blocks_2_adaLN_modulation_1_weight"), val = tensor<fp32, [1536, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(15843200)))];
|
| 35 |
+
tensor<fp32, [512]> flow_net_res_blocks_2_in_ln_bias = const()[name = tensor<string, []>("flow_net_res_blocks_2_in_ln_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(18988992)))];
|
| 36 |
+
tensor<fp32, [512]> flow_net_res_blocks_2_in_ln_weight = const()[name = tensor<string, []>("flow_net_res_blocks_2_in_ln_weight"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(18991104)))];
|
| 37 |
+
tensor<fp32, [512]> flow_net_res_blocks_2_mlp_0_bias = const()[name = tensor<string, []>("flow_net_res_blocks_2_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(18993216)))];
|
| 38 |
+
tensor<fp32, [512, 512]> flow_net_res_blocks_2_mlp_0_weight = const()[name = tensor<string, []>("flow_net_res_blocks_2_mlp_0_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(18995328)))];
|
| 39 |
+
tensor<fp32, [512]> flow_net_res_blocks_2_mlp_2_bias = const()[name = tensor<string, []>("flow_net_res_blocks_2_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(20043968)))];
|
| 40 |
+
tensor<fp32, [512, 512]> flow_net_res_blocks_2_mlp_2_weight = const()[name = tensor<string, []>("flow_net_res_blocks_2_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(20046080)))];
|
| 41 |
+
tensor<fp32, [1536]> flow_net_res_blocks_3_adaLN_modulation_1_bias = const()[name = tensor<string, []>("flow_net_res_blocks_3_adaLN_modulation_1_bias"), val = tensor<fp32, [1536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(21094720)))];
|
| 42 |
+
tensor<fp32, [1536, 512]> flow_net_res_blocks_3_adaLN_modulation_1_weight = const()[name = tensor<string, []>("flow_net_res_blocks_3_adaLN_modulation_1_weight"), val = tensor<fp32, [1536, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(21100928)))];
|
| 43 |
+
tensor<fp32, [512]> flow_net_res_blocks_3_in_ln_bias = const()[name = tensor<string, []>("flow_net_res_blocks_3_in_ln_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(24246720)))];
|
| 44 |
+
tensor<fp32, [512]> flow_net_res_blocks_3_in_ln_weight = const()[name = tensor<string, []>("flow_net_res_blocks_3_in_ln_weight"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(24248832)))];
|
| 45 |
+
tensor<fp32, [512]> flow_net_res_blocks_3_mlp_0_bias = const()[name = tensor<string, []>("flow_net_res_blocks_3_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(24250944)))];
|
| 46 |
+
tensor<fp32, [512, 512]> flow_net_res_blocks_3_mlp_0_weight = const()[name = tensor<string, []>("flow_net_res_blocks_3_mlp_0_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(24253056)))];
|
| 47 |
+
tensor<fp32, [512]> flow_net_res_blocks_3_mlp_2_bias = const()[name = tensor<string, []>("flow_net_res_blocks_3_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(25301696)))];
|
| 48 |
+
tensor<fp32, [512, 512]> flow_net_res_blocks_3_mlp_2_weight = const()[name = tensor<string, []>("flow_net_res_blocks_3_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(25303808)))];
|
| 49 |
+
tensor<fp32, [1536]> flow_net_res_blocks_4_adaLN_modulation_1_bias = const()[name = tensor<string, []>("flow_net_res_blocks_4_adaLN_modulation_1_bias"), val = tensor<fp32, [1536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(26352448)))];
|
| 50 |
+
tensor<fp32, [1536, 512]> flow_net_res_blocks_4_adaLN_modulation_1_weight = const()[name = tensor<string, []>("flow_net_res_blocks_4_adaLN_modulation_1_weight"), val = tensor<fp32, [1536, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(26358656)))];
|
| 51 |
+
tensor<fp32, [512]> flow_net_res_blocks_4_in_ln_bias = const()[name = tensor<string, []>("flow_net_res_blocks_4_in_ln_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(29504448)))];
|
| 52 |
+
tensor<fp32, [512]> flow_net_res_blocks_4_in_ln_weight = const()[name = tensor<string, []>("flow_net_res_blocks_4_in_ln_weight"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(29506560)))];
|
| 53 |
+
tensor<fp32, [512]> flow_net_res_blocks_4_mlp_0_bias = const()[name = tensor<string, []>("flow_net_res_blocks_4_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(29508672)))];
|
| 54 |
+
tensor<fp32, [512, 512]> flow_net_res_blocks_4_mlp_0_weight = const()[name = tensor<string, []>("flow_net_res_blocks_4_mlp_0_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(29510784)))];
|
| 55 |
+
tensor<fp32, [512]> flow_net_res_blocks_4_mlp_2_bias = const()[name = tensor<string, []>("flow_net_res_blocks_4_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(30559424)))];
|
| 56 |
+
tensor<fp32, [512, 512]> flow_net_res_blocks_4_mlp_2_weight = const()[name = tensor<string, []>("flow_net_res_blocks_4_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(30561536)))];
|
| 57 |
+
tensor<fp32, [1536]> flow_net_res_blocks_5_adaLN_modulation_1_bias = const()[name = tensor<string, []>("flow_net_res_blocks_5_adaLN_modulation_1_bias"), val = tensor<fp32, [1536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(31610176)))];
|
| 58 |
+
tensor<fp32, [1536, 512]> flow_net_res_blocks_5_adaLN_modulation_1_weight = const()[name = tensor<string, []>("flow_net_res_blocks_5_adaLN_modulation_1_weight"), val = tensor<fp32, [1536, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(31616384)))];
|
| 59 |
+
tensor<fp32, [512]> flow_net_res_blocks_5_in_ln_bias = const()[name = tensor<string, []>("flow_net_res_blocks_5_in_ln_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(34762176)))];
|
| 60 |
+
tensor<fp32, [512]> flow_net_res_blocks_5_in_ln_weight = const()[name = tensor<string, []>("flow_net_res_blocks_5_in_ln_weight"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(34764288)))];
|
| 61 |
+
tensor<fp32, [512]> flow_net_res_blocks_5_mlp_0_bias = const()[name = tensor<string, []>("flow_net_res_blocks_5_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(34766400)))];
|
| 62 |
+
tensor<fp32, [512, 512]> flow_net_res_blocks_5_mlp_0_weight = const()[name = tensor<string, []>("flow_net_res_blocks_5_mlp_0_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(34768512)))];
|
| 63 |
+
tensor<fp32, [512]> flow_net_res_blocks_5_mlp_2_bias = const()[name = tensor<string, []>("flow_net_res_blocks_5_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(35817152)))];
|
| 64 |
+
tensor<fp32, [512, 512]> flow_net_res_blocks_5_mlp_2_weight = const()[name = tensor<string, []>("flow_net_res_blocks_5_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(35819264)))];
|
| 65 |
+
tensor<fp32, [1024]> flow_net_final_layer_adaLN_modulation_1_bias = const()[name = tensor<string, []>("flow_net_final_layer_adaLN_modulation_1_bias"), val = tensor<fp32, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(36867904)))];
|
| 66 |
+
tensor<fp32, [1024, 512]> flow_net_final_layer_adaLN_modulation_1_weight = const()[name = tensor<string, []>("flow_net_final_layer_adaLN_modulation_1_weight"), val = tensor<fp32, [1024, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(36872064)))];
|
| 67 |
+
tensor<fp32, [32]> flow_net_final_layer_linear_bias = const()[name = tensor<string, []>("flow_net_final_layer_linear_bias"), val = tensor<fp32, [32]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(38969280)))];
|
| 68 |
+
tensor<fp32, [32, 512]> flow_net_final_layer_linear_weight = const()[name = tensor<string, []>("flow_net_final_layer_linear_weight"), val = tensor<fp32, [32, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(38969472)))];
|
| 69 |
+
tensor<int32, []> var_9 = const()[name = tensor<string, []>("op_9"), val = tensor<int32, []>(-1)];
|
| 70 |
+
tensor<fp32, [1, 512]> x_5 = linear(bias = flow_net_input_proj_bias, weight = flow_net_input_proj_weight, x = latent)[name = tensor<string, []>("linear_0")];
|
| 71 |
+
tensor<fp32, [128]> const_0 = const()[name = tensor<string, []>("const_0"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(39035072)))];
|
| 72 |
+
tensor<fp32, [1, 128]> args_1 = mul(x = s, y = const_0)[name = tensor<string, []>("args_1")];
|
| 73 |
+
tensor<fp32, [1, 128]> var_39 = cos(x = args_1)[name = tensor<string, []>("op_39")];
|
| 74 |
+
tensor<fp32, [1, 128]> var_40 = sin(x = args_1)[name = tensor<string, []>("op_40")];
|
| 75 |
+
tensor<bool, []> input_1_interleave_0 = const()[name = tensor<string, []>("input_1_interleave_0"), val = tensor<bool, []>(false)];
|
| 76 |
+
tensor<fp32, [1, 256]> input_1 = concat(axis = var_9, interleave = input_1_interleave_0, values = (var_39, var_40))[name = tensor<string, []>("input_1")];
|
| 77 |
+
tensor<fp32, [1, 512]> input_3 = linear(bias = flow_net_time_embed_0_mlp_0_bias, weight = flow_net_time_embed_0_mlp_0_weight, x = input_1)[name = tensor<string, []>("linear_1")];
|
| 78 |
+
tensor<fp32, [1, 512]> input_5 = silu(x = input_3)[name = tensor<string, []>("input_5")];
|
| 79 |
+
tensor<fp32, [1, 512]> x_1 = linear(bias = flow_net_time_embed_0_mlp_2_bias, weight = flow_net_time_embed_0_mlp_2_weight, x = input_5)[name = tensor<string, []>("linear_2")];
|
| 80 |
+
tensor<int32, [1]> reduce_mean_0_axes_0 = const()[name = tensor<string, []>("reduce_mean_0_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 81 |
+
tensor<bool, []> reduce_mean_0_keep_dims_0 = const()[name = tensor<string, []>("reduce_mean_0_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 82 |
+
tensor<fp32, [1, 1]> reduce_mean_0 = reduce_mean(axes = reduce_mean_0_axes_0, keep_dims = reduce_mean_0_keep_dims_0, x = x_1)[name = tensor<string, []>("reduce_mean_0")];
|
| 83 |
+
tensor<fp32, [1, 512]> sub_0 = sub(x = x_1, y = reduce_mean_0)[name = tensor<string, []>("sub_0")];
|
| 84 |
+
tensor<fp32, [1, 512]> square_0 = square(x = sub_0)[name = tensor<string, []>("square_0")];
|
| 85 |
+
tensor<int32, [1]> reduce_mean_1_axes_0 = const()[name = tensor<string, []>("reduce_mean_1_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 86 |
+
tensor<bool, []> reduce_mean_1_keep_dims_0 = const()[name = tensor<string, []>("reduce_mean_1_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 87 |
+
tensor<fp32, [1, 1]> reduce_mean_1 = reduce_mean(axes = reduce_mean_1_axes_0, keep_dims = reduce_mean_1_keep_dims_0, x = square_0)[name = tensor<string, []>("reduce_mean_1")];
|
| 88 |
+
tensor<fp32, []> real_div_0 = const()[name = tensor<string, []>("real_div_0"), val = tensor<fp32, []>(0x1.00804p+0)];
|
| 89 |
+
tensor<fp32, [1, 1]> mul_0 = mul(x = reduce_mean_1, y = real_div_0)[name = tensor<string, []>("mul_0")];
|
| 90 |
+
tensor<fp32, []> var_56 = const()[name = tensor<string, []>("op_56"), val = tensor<fp32, []>(0x1.4f8b58p-17)];
|
| 91 |
+
tensor<fp32, [1, 1]> var_1 = add(x = mul_0, y = var_56)[name = tensor<string, []>("var_1")];
|
| 92 |
+
tensor<fp32, [512]> const_1 = const()[name = tensor<string, []>("const_1"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(39035648)))];
|
| 93 |
+
tensor<fp32, []> var_59_epsilon_0 = const()[name = tensor<string, []>("op_59_epsilon_0"), val = tensor<fp32, []>(0x1.197998p-40)];
|
| 94 |
+
tensor<fp32, [1, 1]> var_59 = rsqrt(epsilon = var_59_epsilon_0, x = var_1)[name = tensor<string, []>("op_59")];
|
| 95 |
+
tensor<fp32, [1, 512]> var_60 = mul(x = const_1, y = var_59)[name = tensor<string, []>("op_60")];
|
| 96 |
+
tensor<fp32, [1, 512]> var_61 = mul(x = x_1, y = var_60)[name = tensor<string, []>("op_61")];
|
| 97 |
+
tensor<fp32, [1, 128]> args = mul(x = t, y = const_0)[name = tensor<string, []>("args")];
|
| 98 |
+
tensor<fp32, [1, 128]> var_69 = cos(x = args)[name = tensor<string, []>("op_69")];
|
| 99 |
+
tensor<fp32, [1, 128]> var_70 = sin(x = args)[name = tensor<string, []>("op_70")];
|
| 100 |
+
tensor<bool, []> input_7_interleave_0 = const()[name = tensor<string, []>("input_7_interleave_0"), val = tensor<bool, []>(false)];
|
| 101 |
+
tensor<fp32, [1, 256]> input_7 = concat(axis = var_9, interleave = input_7_interleave_0, values = (var_69, var_70))[name = tensor<string, []>("input_7")];
|
| 102 |
+
tensor<fp32, [1, 512]> input_9 = linear(bias = flow_net_time_embed_1_mlp_0_bias, weight = flow_net_time_embed_1_mlp_0_weight, x = input_7)[name = tensor<string, []>("linear_3")];
|
| 103 |
+
tensor<fp32, [1, 512]> input_11 = silu(x = input_9)[name = tensor<string, []>("input_11")];
|
| 104 |
+
tensor<fp32, [1, 512]> x_3 = linear(bias = flow_net_time_embed_1_mlp_2_bias, weight = flow_net_time_embed_1_mlp_2_weight, x = input_11)[name = tensor<string, []>("linear_4")];
|
| 105 |
+
tensor<int32, [1]> reduce_mean_2_axes_0 = const()[name = tensor<string, []>("reduce_mean_2_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 106 |
+
tensor<bool, []> reduce_mean_2_keep_dims_0 = const()[name = tensor<string, []>("reduce_mean_2_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 107 |
+
tensor<fp32, [1, 1]> reduce_mean_2 = reduce_mean(axes = reduce_mean_2_axes_0, keep_dims = reduce_mean_2_keep_dims_0, x = x_3)[name = tensor<string, []>("reduce_mean_2")];
|
| 108 |
+
tensor<fp32, [1, 512]> sub_2 = sub(x = x_3, y = reduce_mean_2)[name = tensor<string, []>("sub_2")];
|
| 109 |
+
tensor<fp32, [1, 512]> square_1 = square(x = sub_2)[name = tensor<string, []>("square_1")];
|
| 110 |
+
tensor<int32, [1]> reduce_mean_3_axes_0 = const()[name = tensor<string, []>("reduce_mean_3_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 111 |
+
tensor<bool, []> reduce_mean_3_keep_dims_0 = const()[name = tensor<string, []>("reduce_mean_3_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 112 |
+
tensor<fp32, [1, 1]> reduce_mean_3 = reduce_mean(axes = reduce_mean_3_axes_0, keep_dims = reduce_mean_3_keep_dims_0, x = square_1)[name = tensor<string, []>("reduce_mean_3")];
|
| 113 |
+
tensor<fp32, []> real_div_1 = const()[name = tensor<string, []>("real_div_1"), val = tensor<fp32, []>(0x1.00804p+0)];
|
| 114 |
+
tensor<fp32, [1, 1]> mul_1 = mul(x = reduce_mean_3, y = real_div_1)[name = tensor<string, []>("mul_1")];
|
| 115 |
+
tensor<fp32, []> var_86 = const()[name = tensor<string, []>("op_86"), val = tensor<fp32, []>(0x1.4f8b58p-17)];
|
| 116 |
+
tensor<fp32, [1, 1]> var_3 = add(x = mul_1, y = var_86)[name = tensor<string, []>("var_3")];
|
| 117 |
+
tensor<fp32, [512]> const_3 = const()[name = tensor<string, []>("const_3"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(39037760)))];
|
| 118 |
+
tensor<fp32, []> var_89_epsilon_0 = const()[name = tensor<string, []>("op_89_epsilon_0"), val = tensor<fp32, []>(0x1.197998p-40)];
|
| 119 |
+
tensor<fp32, [1, 1]> var_89 = rsqrt(epsilon = var_89_epsilon_0, x = var_3)[name = tensor<string, []>("op_89")];
|
| 120 |
+
tensor<fp32, [1, 512]> var_90 = mul(x = const_3, y = var_89)[name = tensor<string, []>("op_90")];
|
| 121 |
+
tensor<fp32, [1, 512]> var_91 = mul(x = x_3, y = var_90)[name = tensor<string, []>("op_91")];
|
| 122 |
+
tensor<fp32, [1, 512]> var_93 = add(x = var_61, y = var_91)[name = tensor<string, []>("op_93")];
|
| 123 |
+
tensor<fp32, []> _inversed_t_combined_y_0 = const()[name = tensor<string, []>("_inversed_t_combined_y_0"), val = tensor<fp32, []>(0x1p-1)];
|
| 124 |
+
tensor<fp32, [1, 512]> _inversed_t_combined = mul(x = var_93, y = _inversed_t_combined_y_0)[name = tensor<string, []>("_inversed_t_combined")];
|
| 125 |
+
tensor<fp32, [1, 512]> c = linear(bias = flow_net_cond_embed_bias, weight = flow_net_cond_embed_weight, x = transformer_out)[name = tensor<string, []>("linear_5")];
|
| 126 |
+
tensor<fp32, [1, 512]> input_13 = add(x = _inversed_t_combined, y = c)[name = tensor<string, []>("input_13")];
|
| 127 |
+
tensor<fp32, [1, 512]> input_15 = silu(x = input_13)[name = tensor<string, []>("input_15")];
|
| 128 |
+
tensor<fp32, [1, 1536]> var_107 = linear(bias = flow_net_res_blocks_0_adaLN_modulation_1_bias, weight = flow_net_res_blocks_0_adaLN_modulation_1_weight, x = input_15)[name = tensor<string, []>("linear_6")];
|
| 129 |
+
tensor<int32, [3]> var_108_split_sizes_0 = const()[name = tensor<string, []>("op_108_split_sizes_0"), val = tensor<int32, [3]>([512, 512, 512])];
|
| 130 |
+
tensor<int32, []> var_108_axis_0 = const()[name = tensor<string, []>("op_108_axis_0"), val = tensor<int32, []>(-1)];
|
| 131 |
+
tensor<fp32, [1, 512]> var_108_0, tensor<fp32, [1, 512]> var_108_1, tensor<fp32, [1, 512]> var_108_2 = split(axis = var_108_axis_0, split_sizes = var_108_split_sizes_0, x = var_107)[name = tensor<string, []>("op_108")];
|
| 132 |
+
tensor<int32, [1]> mean_1_axes_0 = const()[name = tensor<string, []>("mean_1_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 133 |
+
tensor<bool, []> mean_1_keep_dims_0 = const()[name = tensor<string, []>("mean_1_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 134 |
+
tensor<fp32, [1, 1]> mean_1 = reduce_mean(axes = mean_1_axes_0, keep_dims = mean_1_keep_dims_0, x = x_5)[name = tensor<string, []>("mean_1")];
|
| 135 |
+
tensor<fp32, [1, 512]> sub_4 = sub(x = x_5, y = mean_1)[name = tensor<string, []>("sub_4")];
|
| 136 |
+
tensor<fp32, [1, 512]> square_2 = square(x = sub_4)[name = tensor<string, []>("square_2")];
|
| 137 |
+
tensor<int32, [1]> reduce_mean_5_axes_0 = const()[name = tensor<string, []>("reduce_mean_5_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 138 |
+
tensor<bool, []> reduce_mean_5_keep_dims_0 = const()[name = tensor<string, []>("reduce_mean_5_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 139 |
+
tensor<fp32, [1, 1]> reduce_mean_5 = reduce_mean(axes = reduce_mean_5_axes_0, keep_dims = reduce_mean_5_keep_dims_0, x = square_2)[name = tensor<string, []>("reduce_mean_5")];
|
| 140 |
+
tensor<fp32, []> var_118 = const()[name = tensor<string, []>("op_118"), val = tensor<fp32, []>(0x1.0c6f7ap-20)];
|
| 141 |
+
tensor<fp32, [1, 1]> var_119 = add(x = reduce_mean_5, y = var_118)[name = tensor<string, []>("op_119")];
|
| 142 |
+
tensor<fp32, [1, 1]> var_120 = sqrt(x = var_119)[name = tensor<string, []>("op_120")];
|
| 143 |
+
tensor<fp32, [1, 512]> x_7 = real_div(x = sub_4, y = var_120)[name = tensor<string, []>("x_7")];
|
| 144 |
+
tensor<fp32, [1, 512]> var_122 = mul(x = x_7, y = flow_net_res_blocks_0_in_ln_weight)[name = tensor<string, []>("op_122")];
|
| 145 |
+
tensor<fp32, [1, 512]> x_9 = add(x = var_122, y = flow_net_res_blocks_0_in_ln_bias)[name = tensor<string, []>("x_9")];
|
| 146 |
+
tensor<fp32, []> var_124_promoted = const()[name = tensor<string, []>("op_124_promoted"), val = tensor<fp32, []>(0x1p+0)];
|
| 147 |
+
tensor<fp32, [1, 512]> var_125 = add(x = var_108_1, y = var_124_promoted)[name = tensor<string, []>("op_125")];
|
| 148 |
+
tensor<fp32, [1, 512]> var_126 = mul(x = x_9, y = var_125)[name = tensor<string, []>("op_126")];
|
| 149 |
+
tensor<fp32, [1, 512]> input_17 = add(x = var_126, y = var_108_0)[name = tensor<string, []>("input_17")];
|
| 150 |
+
tensor<fp32, [1, 512]> input_19 = linear(bias = flow_net_res_blocks_0_mlp_0_bias, weight = flow_net_res_blocks_0_mlp_0_weight, x = input_17)[name = tensor<string, []>("linear_7")];
|
| 151 |
+
tensor<fp32, [1, 512]> input_21 = silu(x = input_19)[name = tensor<string, []>("input_21")];
|
| 152 |
+
tensor<fp32, [1, 512]> h_1 = linear(bias = flow_net_res_blocks_0_mlp_2_bias, weight = flow_net_res_blocks_0_mlp_2_weight, x = input_21)[name = tensor<string, []>("linear_8")];
|
| 153 |
+
tensor<fp32, [1, 512]> var_137 = mul(x = var_108_2, y = h_1)[name = tensor<string, []>("op_137")];
|
| 154 |
+
tensor<fp32, [1, 512]> x_11 = add(x = x_5, y = var_137)[name = tensor<string, []>("x_11")];
|
| 155 |
+
tensor<fp32, [1, 1536]> var_146 = linear(bias = flow_net_res_blocks_1_adaLN_modulation_1_bias, weight = flow_net_res_blocks_1_adaLN_modulation_1_weight, x = input_15)[name = tensor<string, []>("linear_9")];
|
| 156 |
+
tensor<int32, [3]> var_147_split_sizes_0 = const()[name = tensor<string, []>("op_147_split_sizes_0"), val = tensor<int32, [3]>([512, 512, 512])];
|
| 157 |
+
tensor<int32, []> var_147_axis_0 = const()[name = tensor<string, []>("op_147_axis_0"), val = tensor<int32, []>(-1)];
|
| 158 |
+
tensor<fp32, [1, 512]> var_147_0, tensor<fp32, [1, 512]> var_147_1, tensor<fp32, [1, 512]> var_147_2 = split(axis = var_147_axis_0, split_sizes = var_147_split_sizes_0, x = var_146)[name = tensor<string, []>("op_147")];
|
| 159 |
+
tensor<int32, [1]> mean_3_axes_0 = const()[name = tensor<string, []>("mean_3_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 160 |
+
tensor<bool, []> mean_3_keep_dims_0 = const()[name = tensor<string, []>("mean_3_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 161 |
+
tensor<fp32, [1, 1]> mean_3 = reduce_mean(axes = mean_3_axes_0, keep_dims = mean_3_keep_dims_0, x = x_11)[name = tensor<string, []>("mean_3")];
|
| 162 |
+
tensor<fp32, [1, 512]> sub_5 = sub(x = x_11, y = mean_3)[name = tensor<string, []>("sub_5")];
|
| 163 |
+
tensor<fp32, [1, 512]> square_3 = square(x = sub_5)[name = tensor<string, []>("square_3")];
|
| 164 |
+
tensor<int32, [1]> reduce_mean_7_axes_0 = const()[name = tensor<string, []>("reduce_mean_7_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 165 |
+
tensor<bool, []> reduce_mean_7_keep_dims_0 = const()[name = tensor<string, []>("reduce_mean_7_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 166 |
+
tensor<fp32, [1, 1]> reduce_mean_7 = reduce_mean(axes = reduce_mean_7_axes_0, keep_dims = reduce_mean_7_keep_dims_0, x = square_3)[name = tensor<string, []>("reduce_mean_7")];
|
| 167 |
+
tensor<fp32, []> var_157 = const()[name = tensor<string, []>("op_157"), val = tensor<fp32, []>(0x1.0c6f7ap-20)];
|
| 168 |
+
tensor<fp32, [1, 1]> var_158 = add(x = reduce_mean_7, y = var_157)[name = tensor<string, []>("op_158")];
|
| 169 |
+
tensor<fp32, [1, 1]> var_159 = sqrt(x = var_158)[name = tensor<string, []>("op_159")];
|
| 170 |
+
tensor<fp32, [1, 512]> x_13 = real_div(x = sub_5, y = var_159)[name = tensor<string, []>("x_13")];
|
| 171 |
+
tensor<fp32, [1, 512]> var_161 = mul(x = x_13, y = flow_net_res_blocks_1_in_ln_weight)[name = tensor<string, []>("op_161")];
|
| 172 |
+
tensor<fp32, [1, 512]> x_15 = add(x = var_161, y = flow_net_res_blocks_1_in_ln_bias)[name = tensor<string, []>("x_15")];
|
| 173 |
+
tensor<fp32, []> var_163_promoted = const()[name = tensor<string, []>("op_163_promoted"), val = tensor<fp32, []>(0x1p+0)];
|
| 174 |
+
tensor<fp32, [1, 512]> var_164 = add(x = var_147_1, y = var_163_promoted)[name = tensor<string, []>("op_164")];
|
| 175 |
+
tensor<fp32, [1, 512]> var_165 = mul(x = x_15, y = var_164)[name = tensor<string, []>("op_165")];
|
| 176 |
+
tensor<fp32, [1, 512]> input_25 = add(x = var_165, y = var_147_0)[name = tensor<string, []>("input_25")];
|
| 177 |
+
tensor<fp32, [1, 512]> input_27 = linear(bias = flow_net_res_blocks_1_mlp_0_bias, weight = flow_net_res_blocks_1_mlp_0_weight, x = input_25)[name = tensor<string, []>("linear_10")];
|
| 178 |
+
tensor<fp32, [1, 512]> input_29 = silu(x = input_27)[name = tensor<string, []>("input_29")];
|
| 179 |
+
tensor<fp32, [1, 512]> h_3 = linear(bias = flow_net_res_blocks_1_mlp_2_bias, weight = flow_net_res_blocks_1_mlp_2_weight, x = input_29)[name = tensor<string, []>("linear_11")];
|
| 180 |
+
tensor<fp32, [1, 512]> var_176 = mul(x = var_147_2, y = h_3)[name = tensor<string, []>("op_176")];
|
| 181 |
+
tensor<fp32, [1, 512]> x_17 = add(x = x_11, y = var_176)[name = tensor<string, []>("x_17")];
|
| 182 |
+
tensor<fp32, [1, 1536]> var_185 = linear(bias = flow_net_res_blocks_2_adaLN_modulation_1_bias, weight = flow_net_res_blocks_2_adaLN_modulation_1_weight, x = input_15)[name = tensor<string, []>("linear_12")];
|
| 183 |
+
tensor<int32, [3]> var_186_split_sizes_0 = const()[name = tensor<string, []>("op_186_split_sizes_0"), val = tensor<int32, [3]>([512, 512, 512])];
|
| 184 |
+
tensor<int32, []> var_186_axis_0 = const()[name = tensor<string, []>("op_186_axis_0"), val = tensor<int32, []>(-1)];
|
| 185 |
+
tensor<fp32, [1, 512]> var_186_0, tensor<fp32, [1, 512]> var_186_1, tensor<fp32, [1, 512]> var_186_2 = split(axis = var_186_axis_0, split_sizes = var_186_split_sizes_0, x = var_185)[name = tensor<string, []>("op_186")];
|
| 186 |
+
tensor<int32, [1]> mean_5_axes_0 = const()[name = tensor<string, []>("mean_5_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 187 |
+
tensor<bool, []> mean_5_keep_dims_0 = const()[name = tensor<string, []>("mean_5_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 188 |
+
tensor<fp32, [1, 1]> mean_5 = reduce_mean(axes = mean_5_axes_0, keep_dims = mean_5_keep_dims_0, x = x_17)[name = tensor<string, []>("mean_5")];
|
| 189 |
+
tensor<fp32, [1, 512]> sub_6 = sub(x = x_17, y = mean_5)[name = tensor<string, []>("sub_6")];
|
| 190 |
+
tensor<fp32, [1, 512]> square_4 = square(x = sub_6)[name = tensor<string, []>("square_4")];
|
| 191 |
+
tensor<int32, [1]> reduce_mean_9_axes_0 = const()[name = tensor<string, []>("reduce_mean_9_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 192 |
+
tensor<bool, []> reduce_mean_9_keep_dims_0 = const()[name = tensor<string, []>("reduce_mean_9_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 193 |
+
tensor<fp32, [1, 1]> reduce_mean_9 = reduce_mean(axes = reduce_mean_9_axes_0, keep_dims = reduce_mean_9_keep_dims_0, x = square_4)[name = tensor<string, []>("reduce_mean_9")];
|
| 194 |
+
tensor<fp32, []> var_196 = const()[name = tensor<string, []>("op_196"), val = tensor<fp32, []>(0x1.0c6f7ap-20)];
|
| 195 |
+
tensor<fp32, [1, 1]> var_197 = add(x = reduce_mean_9, y = var_196)[name = tensor<string, []>("op_197")];
|
| 196 |
+
tensor<fp32, [1, 1]> var_198 = sqrt(x = var_197)[name = tensor<string, []>("op_198")];
|
| 197 |
+
tensor<fp32, [1, 512]> x_19 = real_div(x = sub_6, y = var_198)[name = tensor<string, []>("x_19")];
|
| 198 |
+
tensor<fp32, [1, 512]> var_200 = mul(x = x_19, y = flow_net_res_blocks_2_in_ln_weight)[name = tensor<string, []>("op_200")];
|
| 199 |
+
tensor<fp32, [1, 512]> x_21 = add(x = var_200, y = flow_net_res_blocks_2_in_ln_bias)[name = tensor<string, []>("x_21")];
|
| 200 |
+
tensor<fp32, []> var_202_promoted = const()[name = tensor<string, []>("op_202_promoted"), val = tensor<fp32, []>(0x1p+0)];
|
| 201 |
+
tensor<fp32, [1, 512]> var_203 = add(x = var_186_1, y = var_202_promoted)[name = tensor<string, []>("op_203")];
|
| 202 |
+
tensor<fp32, [1, 512]> var_204 = mul(x = x_21, y = var_203)[name = tensor<string, []>("op_204")];
|
| 203 |
+
tensor<fp32, [1, 512]> input_33 = add(x = var_204, y = var_186_0)[name = tensor<string, []>("input_33")];
|
| 204 |
+
tensor<fp32, [1, 512]> input_35 = linear(bias = flow_net_res_blocks_2_mlp_0_bias, weight = flow_net_res_blocks_2_mlp_0_weight, x = input_33)[name = tensor<string, []>("linear_13")];
|
| 205 |
+
tensor<fp32, [1, 512]> input_37 = silu(x = input_35)[name = tensor<string, []>("input_37")];
|
| 206 |
+
tensor<fp32, [1, 512]> h_5 = linear(bias = flow_net_res_blocks_2_mlp_2_bias, weight = flow_net_res_blocks_2_mlp_2_weight, x = input_37)[name = tensor<string, []>("linear_14")];
|
| 207 |
+
tensor<fp32, [1, 512]> var_215 = mul(x = var_186_2, y = h_5)[name = tensor<string, []>("op_215")];
|
| 208 |
+
tensor<fp32, [1, 512]> x_23 = add(x = x_17, y = var_215)[name = tensor<string, []>("x_23")];
|
| 209 |
+
tensor<fp32, [1, 1536]> var_224 = linear(bias = flow_net_res_blocks_3_adaLN_modulation_1_bias, weight = flow_net_res_blocks_3_adaLN_modulation_1_weight, x = input_15)[name = tensor<string, []>("linear_15")];
|
| 210 |
+
tensor<int32, [3]> var_225_split_sizes_0 = const()[name = tensor<string, []>("op_225_split_sizes_0"), val = tensor<int32, [3]>([512, 512, 512])];
|
| 211 |
+
tensor<int32, []> var_225_axis_0 = const()[name = tensor<string, []>("op_225_axis_0"), val = tensor<int32, []>(-1)];
|
| 212 |
+
tensor<fp32, [1, 512]> var_225_0, tensor<fp32, [1, 512]> var_225_1, tensor<fp32, [1, 512]> var_225_2 = split(axis = var_225_axis_0, split_sizes = var_225_split_sizes_0, x = var_224)[name = tensor<string, []>("op_225")];
|
| 213 |
+
tensor<int32, [1]> mean_7_axes_0 = const()[name = tensor<string, []>("mean_7_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 214 |
+
tensor<bool, []> mean_7_keep_dims_0 = const()[name = tensor<string, []>("mean_7_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 215 |
+
tensor<fp32, [1, 1]> mean_7 = reduce_mean(axes = mean_7_axes_0, keep_dims = mean_7_keep_dims_0, x = x_23)[name = tensor<string, []>("mean_7")];
|
| 216 |
+
tensor<fp32, [1, 512]> sub_7 = sub(x = x_23, y = mean_7)[name = tensor<string, []>("sub_7")];
|
| 217 |
+
tensor<fp32, [1, 512]> square_5 = square(x = sub_7)[name = tensor<string, []>("square_5")];
|
| 218 |
+
tensor<int32, [1]> reduce_mean_11_axes_0 = const()[name = tensor<string, []>("reduce_mean_11_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 219 |
+
tensor<bool, []> reduce_mean_11_keep_dims_0 = const()[name = tensor<string, []>("reduce_mean_11_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 220 |
+
tensor<fp32, [1, 1]> reduce_mean_11 = reduce_mean(axes = reduce_mean_11_axes_0, keep_dims = reduce_mean_11_keep_dims_0, x = square_5)[name = tensor<string, []>("reduce_mean_11")];
|
| 221 |
+
tensor<fp32, []> var_235 = const()[name = tensor<string, []>("op_235"), val = tensor<fp32, []>(0x1.0c6f7ap-20)];
|
| 222 |
+
tensor<fp32, [1, 1]> var_236 = add(x = reduce_mean_11, y = var_235)[name = tensor<string, []>("op_236")];
|
| 223 |
+
tensor<fp32, [1, 1]> var_237 = sqrt(x = var_236)[name = tensor<string, []>("op_237")];
|
| 224 |
+
tensor<fp32, [1, 512]> x_25 = real_div(x = sub_7, y = var_237)[name = tensor<string, []>("x_25")];
|
| 225 |
+
tensor<fp32, [1, 512]> var_239 = mul(x = x_25, y = flow_net_res_blocks_3_in_ln_weight)[name = tensor<string, []>("op_239")];
|
| 226 |
+
tensor<fp32, [1, 512]> x_27 = add(x = var_239, y = flow_net_res_blocks_3_in_ln_bias)[name = tensor<string, []>("x_27")];
|
| 227 |
+
tensor<fp32, []> var_241_promoted = const()[name = tensor<string, []>("op_241_promoted"), val = tensor<fp32, []>(0x1p+0)];
|
| 228 |
+
tensor<fp32, [1, 512]> var_242 = add(x = var_225_1, y = var_241_promoted)[name = tensor<string, []>("op_242")];
|
| 229 |
+
tensor<fp32, [1, 512]> var_243 = mul(x = x_27, y = var_242)[name = tensor<string, []>("op_243")];
|
| 230 |
+
tensor<fp32, [1, 512]> input_41 = add(x = var_243, y = var_225_0)[name = tensor<string, []>("input_41")];
|
| 231 |
+
tensor<fp32, [1, 512]> input_43 = linear(bias = flow_net_res_blocks_3_mlp_0_bias, weight = flow_net_res_blocks_3_mlp_0_weight, x = input_41)[name = tensor<string, []>("linear_16")];
|
| 232 |
+
tensor<fp32, [1, 512]> input_45 = silu(x = input_43)[name = tensor<string, []>("input_45")];
|
| 233 |
+
tensor<fp32, [1, 512]> h_7 = linear(bias = flow_net_res_blocks_3_mlp_2_bias, weight = flow_net_res_blocks_3_mlp_2_weight, x = input_45)[name = tensor<string, []>("linear_17")];
|
| 234 |
+
tensor<fp32, [1, 512]> var_254 = mul(x = var_225_2, y = h_7)[name = tensor<string, []>("op_254")];
|
| 235 |
+
tensor<fp32, [1, 512]> x_29 = add(x = x_23, y = var_254)[name = tensor<string, []>("x_29")];
|
| 236 |
+
tensor<fp32, [1, 1536]> var_263 = linear(bias = flow_net_res_blocks_4_adaLN_modulation_1_bias, weight = flow_net_res_blocks_4_adaLN_modulation_1_weight, x = input_15)[name = tensor<string, []>("linear_18")];
|
| 237 |
+
tensor<int32, [3]> var_264_split_sizes_0 = const()[name = tensor<string, []>("op_264_split_sizes_0"), val = tensor<int32, [3]>([512, 512, 512])];
|
| 238 |
+
tensor<int32, []> var_264_axis_0 = const()[name = tensor<string, []>("op_264_axis_0"), val = tensor<int32, []>(-1)];
|
| 239 |
+
tensor<fp32, [1, 512]> var_264_0, tensor<fp32, [1, 512]> var_264_1, tensor<fp32, [1, 512]> var_264_2 = split(axis = var_264_axis_0, split_sizes = var_264_split_sizes_0, x = var_263)[name = tensor<string, []>("op_264")];
|
| 240 |
+
tensor<int32, [1]> mean_9_axes_0 = const()[name = tensor<string, []>("mean_9_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 241 |
+
tensor<bool, []> mean_9_keep_dims_0 = const()[name = tensor<string, []>("mean_9_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 242 |
+
tensor<fp32, [1, 1]> mean_9 = reduce_mean(axes = mean_9_axes_0, keep_dims = mean_9_keep_dims_0, x = x_29)[name = tensor<string, []>("mean_9")];
|
| 243 |
+
tensor<fp32, [1, 512]> sub_8 = sub(x = x_29, y = mean_9)[name = tensor<string, []>("sub_8")];
|
| 244 |
+
tensor<fp32, [1, 512]> square_6 = square(x = sub_8)[name = tensor<string, []>("square_6")];
|
| 245 |
+
tensor<int32, [1]> reduce_mean_13_axes_0 = const()[name = tensor<string, []>("reduce_mean_13_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 246 |
+
tensor<bool, []> reduce_mean_13_keep_dims_0 = const()[name = tensor<string, []>("reduce_mean_13_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 247 |
+
tensor<fp32, [1, 1]> reduce_mean_13 = reduce_mean(axes = reduce_mean_13_axes_0, keep_dims = reduce_mean_13_keep_dims_0, x = square_6)[name = tensor<string, []>("reduce_mean_13")];
|
| 248 |
+
tensor<fp32, []> var_274 = const()[name = tensor<string, []>("op_274"), val = tensor<fp32, []>(0x1.0c6f7ap-20)];
|
| 249 |
+
tensor<fp32, [1, 1]> var_275 = add(x = reduce_mean_13, y = var_274)[name = tensor<string, []>("op_275")];
|
| 250 |
+
tensor<fp32, [1, 1]> var_276 = sqrt(x = var_275)[name = tensor<string, []>("op_276")];
|
| 251 |
+
tensor<fp32, [1, 512]> x_31 = real_div(x = sub_8, y = var_276)[name = tensor<string, []>("x_31")];
|
| 252 |
+
tensor<fp32, [1, 512]> var_278 = mul(x = x_31, y = flow_net_res_blocks_4_in_ln_weight)[name = tensor<string, []>("op_278")];
|
| 253 |
+
tensor<fp32, [1, 512]> x_33 = add(x = var_278, y = flow_net_res_blocks_4_in_ln_bias)[name = tensor<string, []>("x_33")];
|
| 254 |
+
tensor<fp32, []> var_280_promoted = const()[name = tensor<string, []>("op_280_promoted"), val = tensor<fp32, []>(0x1p+0)];
|
| 255 |
+
tensor<fp32, [1, 512]> var_281 = add(x = var_264_1, y = var_280_promoted)[name = tensor<string, []>("op_281")];
|
| 256 |
+
tensor<fp32, [1, 512]> var_282 = mul(x = x_33, y = var_281)[name = tensor<string, []>("op_282")];
|
| 257 |
+
tensor<fp32, [1, 512]> input_49 = add(x = var_282, y = var_264_0)[name = tensor<string, []>("input_49")];
|
| 258 |
+
tensor<fp32, [1, 512]> input_51 = linear(bias = flow_net_res_blocks_4_mlp_0_bias, weight = flow_net_res_blocks_4_mlp_0_weight, x = input_49)[name = tensor<string, []>("linear_19")];
|
| 259 |
+
tensor<fp32, [1, 512]> input_53 = silu(x = input_51)[name = tensor<string, []>("input_53")];
|
| 260 |
+
tensor<fp32, [1, 512]> h_9 = linear(bias = flow_net_res_blocks_4_mlp_2_bias, weight = flow_net_res_blocks_4_mlp_2_weight, x = input_53)[name = tensor<string, []>("linear_20")];
|
| 261 |
+
tensor<fp32, [1, 512]> var_293 = mul(x = var_264_2, y = h_9)[name = tensor<string, []>("op_293")];
|
| 262 |
+
tensor<fp32, [1, 512]> x_35 = add(x = x_29, y = var_293)[name = tensor<string, []>("x_35")];
|
| 263 |
+
tensor<fp32, [1, 1536]> var_302 = linear(bias = flow_net_res_blocks_5_adaLN_modulation_1_bias, weight = flow_net_res_blocks_5_adaLN_modulation_1_weight, x = input_15)[name = tensor<string, []>("linear_21")];
|
| 264 |
+
tensor<int32, [3]> var_303_split_sizes_0 = const()[name = tensor<string, []>("op_303_split_sizes_0"), val = tensor<int32, [3]>([512, 512, 512])];
|
| 265 |
+
tensor<int32, []> var_303_axis_0 = const()[name = tensor<string, []>("op_303_axis_0"), val = tensor<int32, []>(-1)];
|
| 266 |
+
tensor<fp32, [1, 512]> var_303_0, tensor<fp32, [1, 512]> var_303_1, tensor<fp32, [1, 512]> var_303_2 = split(axis = var_303_axis_0, split_sizes = var_303_split_sizes_0, x = var_302)[name = tensor<string, []>("op_303")];
|
| 267 |
+
tensor<int32, [1]> mean_11_axes_0 = const()[name = tensor<string, []>("mean_11_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 268 |
+
tensor<bool, []> mean_11_keep_dims_0 = const()[name = tensor<string, []>("mean_11_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 269 |
+
tensor<fp32, [1, 1]> mean_11 = reduce_mean(axes = mean_11_axes_0, keep_dims = mean_11_keep_dims_0, x = x_35)[name = tensor<string, []>("mean_11")];
|
| 270 |
+
tensor<fp32, [1, 512]> sub_9 = sub(x = x_35, y = mean_11)[name = tensor<string, []>("sub_9")];
|
| 271 |
+
tensor<fp32, [1, 512]> square_7 = square(x = sub_9)[name = tensor<string, []>("square_7")];
|
| 272 |
+
tensor<int32, [1]> reduce_mean_15_axes_0 = const()[name = tensor<string, []>("reduce_mean_15_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 273 |
+
tensor<bool, []> reduce_mean_15_keep_dims_0 = const()[name = tensor<string, []>("reduce_mean_15_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 274 |
+
tensor<fp32, [1, 1]> reduce_mean_15 = reduce_mean(axes = reduce_mean_15_axes_0, keep_dims = reduce_mean_15_keep_dims_0, x = square_7)[name = tensor<string, []>("reduce_mean_15")];
|
| 275 |
+
tensor<fp32, []> var_313 = const()[name = tensor<string, []>("op_313"), val = tensor<fp32, []>(0x1.0c6f7ap-20)];
|
| 276 |
+
tensor<fp32, [1, 1]> var_314 = add(x = reduce_mean_15, y = var_313)[name = tensor<string, []>("op_314")];
|
| 277 |
+
tensor<fp32, [1, 1]> var_315 = sqrt(x = var_314)[name = tensor<string, []>("op_315")];
|
| 278 |
+
tensor<fp32, [1, 512]> x_37 = real_div(x = sub_9, y = var_315)[name = tensor<string, []>("x_37")];
|
| 279 |
+
tensor<fp32, [1, 512]> var_317 = mul(x = x_37, y = flow_net_res_blocks_5_in_ln_weight)[name = tensor<string, []>("op_317")];
|
| 280 |
+
tensor<fp32, [1, 512]> x_39 = add(x = var_317, y = flow_net_res_blocks_5_in_ln_bias)[name = tensor<string, []>("x_39")];
|
| 281 |
+
tensor<fp32, []> var_319_promoted = const()[name = tensor<string, []>("op_319_promoted"), val = tensor<fp32, []>(0x1p+0)];
|
| 282 |
+
tensor<fp32, [1, 512]> var_320 = add(x = var_303_1, y = var_319_promoted)[name = tensor<string, []>("op_320")];
|
| 283 |
+
tensor<fp32, [1, 512]> var_321 = mul(x = x_39, y = var_320)[name = tensor<string, []>("op_321")];
|
| 284 |
+
tensor<fp32, [1, 512]> input_57 = add(x = var_321, y = var_303_0)[name = tensor<string, []>("input_57")];
|
| 285 |
+
tensor<fp32, [1, 512]> input_59 = linear(bias = flow_net_res_blocks_5_mlp_0_bias, weight = flow_net_res_blocks_5_mlp_0_weight, x = input_57)[name = tensor<string, []>("linear_22")];
|
| 286 |
+
tensor<fp32, [1, 512]> input_61 = silu(x = input_59)[name = tensor<string, []>("input_61")];
|
| 287 |
+
tensor<fp32, [1, 512]> h = linear(bias = flow_net_res_blocks_5_mlp_2_bias, weight = flow_net_res_blocks_5_mlp_2_weight, x = input_61)[name = tensor<string, []>("linear_23")];
|
| 288 |
+
tensor<fp32, [1, 512]> var_332 = mul(x = var_303_2, y = h)[name = tensor<string, []>("op_332")];
|
| 289 |
+
tensor<fp32, [1, 512]> x_41 = add(x = x_35, y = var_332)[name = tensor<string, []>("x_41")];
|
| 290 |
+
tensor<fp32, [1, 1024]> var_340 = linear(bias = flow_net_final_layer_adaLN_modulation_1_bias, weight = flow_net_final_layer_adaLN_modulation_1_weight, x = input_15)[name = tensor<string, []>("linear_24")];
|
| 291 |
+
tensor<int32, [2]> var_341_split_sizes_0 = const()[name = tensor<string, []>("op_341_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
|
| 292 |
+
tensor<int32, []> var_341_axis_0 = const()[name = tensor<string, []>("op_341_axis_0"), val = tensor<int32, []>(-1)];
|
| 293 |
+
tensor<fp32, [1, 512]> var_341_0, tensor<fp32, [1, 512]> var_341_1 = split(axis = var_341_axis_0, split_sizes = var_341_split_sizes_0, x = var_340)[name = tensor<string, []>("op_341")];
|
| 294 |
+
tensor<int32, [1]> mean_axes_0 = const()[name = tensor<string, []>("mean_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 295 |
+
tensor<bool, []> mean_keep_dims_0 = const()[name = tensor<string, []>("mean_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 296 |
+
tensor<fp32, [1, 1]> mean = reduce_mean(axes = mean_axes_0, keep_dims = mean_keep_dims_0, x = x_41)[name = tensor<string, []>("mean")];
|
| 297 |
+
tensor<fp32, [1, 512]> sub_10 = sub(x = x_41, y = mean)[name = tensor<string, []>("sub_10")];
|
| 298 |
+
tensor<fp32, [1, 512]> square_8 = square(x = sub_10)[name = tensor<string, []>("square_8")];
|
| 299 |
+
tensor<int32, [1]> reduce_mean_17_axes_0 = const()[name = tensor<string, []>("reduce_mean_17_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 300 |
+
tensor<bool, []> reduce_mean_17_keep_dims_0 = const()[name = tensor<string, []>("reduce_mean_17_keep_dims_0"), val = tensor<bool, []>(true)];
|
| 301 |
+
tensor<fp32, [1, 1]> reduce_mean_17 = reduce_mean(axes = reduce_mean_17_axes_0, keep_dims = reduce_mean_17_keep_dims_0, x = square_8)[name = tensor<string, []>("reduce_mean_17")];
|
| 302 |
+
tensor<fp32, []> var_348 = const()[name = tensor<string, []>("op_348"), val = tensor<fp32, []>(0x1.0c6f7ap-20)];
|
| 303 |
+
tensor<fp32, [1, 1]> var_349 = add(x = reduce_mean_17, y = var_348)[name = tensor<string, []>("op_349")];
|
| 304 |
+
tensor<fp32, [1, 1]> var_350 = sqrt(x = var_349)[name = tensor<string, []>("op_350")];
|
| 305 |
+
tensor<fp32, [1, 512]> x = real_div(x = sub_10, y = var_350)[name = tensor<string, []>("x")];
|
| 306 |
+
tensor<fp32, []> var_352_promoted = const()[name = tensor<string, []>("op_352_promoted"), val = tensor<fp32, []>(0x1p+0)];
|
| 307 |
+
tensor<fp32, [1, 512]> var_353 = add(x = var_341_1, y = var_352_promoted)[name = tensor<string, []>("op_353")];
|
| 308 |
+
tensor<fp32, [1, 512]> var_354 = mul(x = x, y = var_353)[name = tensor<string, []>("op_354")];
|
| 309 |
+
tensor<fp32, [1, 512]> input = add(x = var_354, y = var_341_0)[name = tensor<string, []>("input")];
|
| 310 |
+
tensor<fp32, [1, 32]> var_358 = linear(bias = flow_net_final_layer_linear_bias, weight = flow_net_final_layer_linear_weight, x = input)[name = tensor<string, []>("linear_25")];
|
| 311 |
+
} -> (var_358);
|
| 312 |
+
}
|
v2/english/flow_decoder.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0e03e2fccbc1181c5bfb50d9e60a9869c268426309f0b9ce9382dc72ed137ffd
|
| 3 |
+
size 39039872
|
v2/english/flow_decoder.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dcd7366db0d34328e314169b72bef060e0f598a7c00c78a3c58e84a5cf975f1f
|
| 3 |
+
size 43299
|