neonforestmist commited on
Commit
4ac7126
·
verified ·
1 Parent(s): 77feba6

Place v2 Core ML resources at manifest root

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. Resources/SafetyChecker.mlmodelc/analytics/coremldata.bin +0 -3
  2. Resources/SafetyChecker.mlmodelc/coremldata.bin +0 -3
  3. Resources/SafetyChecker.mlmodelc/metadata.json +0 -130
  4. Resources/SafetyChecker.mlmodelc/model.mil +0 -0
  5. Resources/SafetyChecker.mlmodelc/weights/weight.bin +0 -3
  6. Resources/TextEncoder.mlmodelc/analytics/coremldata.bin +0 -3
  7. Resources/TextEncoder.mlmodelc/coremldata.bin +0 -3
  8. Resources/TextEncoder.mlmodelc/metadata.json +0 -90
  9. Resources/TextEncoder.mlmodelc/model.mil +0 -0
  10. Resources/TextEncoder.mlmodelc/weights/weight.bin +0 -3
  11. Resources/UnetChunk1.mlmodelc/analytics/coremldata.bin +0 -3
  12. Resources/UnetChunk1.mlmodelc/coremldata.bin +0 -3
  13. Resources/UnetChunk1.mlmodelc/metadata.json +0 -180
  14. Resources/UnetChunk1.mlmodelc/model.mil +0 -0
  15. Resources/UnetChunk1.mlmodelc/weights/weight.bin +0 -3
  16. Resources/UnetChunk2.mlmodelc/analytics/coremldata.bin +0 -3
  17. Resources/UnetChunk2.mlmodelc/coremldata.bin +0 -3
  18. Resources/UnetChunk2.mlmodelc/metadata.json +0 -168
  19. Resources/UnetChunk2.mlmodelc/model.mil +0 -0
  20. Resources/UnetChunk2.mlmodelc/weights/weight.bin +0 -3
  21. Resources/VAEDecoder.mlmodelc/analytics/coremldata.bin +0 -3
  22. Resources/VAEDecoder.mlmodelc/coremldata.bin +0 -3
  23. Resources/VAEDecoder.mlmodelc/metadata.json +0 -80
  24. Resources/VAEDecoder.mlmodelc/model.mil +0 -0
  25. Resources/VAEDecoder.mlmodelc/weights/weight.bin +0 -3
  26. Resources/VAEEncoder.mlmodelc/analytics/coremldata.bin +0 -3
  27. Resources/VAEEncoder.mlmodelc/coremldata.bin +0 -3
  28. Resources/VAEEncoder.mlmodelc/metadata.json +0 -80
  29. Resources/VAEEncoder.mlmodelc/model.mil +0 -0
  30. Resources/VAEEncoder.mlmodelc/weights/weight.bin +0 -3
  31. Resources/merges.txt +0 -0
  32. Resources/vocab.json +0 -0
  33. SafetyChecker.mlmodelc/analytics/coremldata.bin +1 -1
  34. SafetyChecker.mlmodelc/coremldata.bin +2 -2
  35. SafetyChecker.mlmodelc/metadata.json +4 -4
  36. TextEncoder.mlmodelc/analytics/coremldata.bin +1 -1
  37. TextEncoder.mlmodelc/coremldata.bin +2 -2
  38. TextEncoder.mlmodelc/metadata.json +6 -6
  39. TextEncoder.mlmodelc/model.mil +0 -0
  40. TextEncoder.mlmodelc/weights/weight.bin +2 -2
  41. UnetChunk1.mlmodelc/analytics/coremldata.bin +1 -1
  42. UnetChunk1.mlmodelc/coremldata.bin +1 -1
  43. UnetChunk1.mlmodelc/metadata.json +19 -19
  44. UnetChunk1.mlmodelc/model.mil +13 -13
  45. UnetChunk1.mlmodelc/weights/weight.bin +1 -1
  46. UnetChunk2.mlmodelc/analytics/coremldata.bin +1 -1
  47. UnetChunk2.mlmodelc/coremldata.bin +1 -1
  48. UnetChunk2.mlmodelc/metadata.json +24 -24
  49. UnetChunk2.mlmodelc/model.mil +30 -30
  50. UnetChunk2.mlmodelc/weights/weight.bin +1 -1
Resources/SafetyChecker.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:8211508a22331e6868fa9d207c6532ae5fc11791e8ceac6bf6fa4c7dd89004e8
3
- size 243
 
 
 
 
Resources/SafetyChecker.mlmodelc/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:f57494e9bc01a089c73ba6b1b9554e1a3df155efbf7cf36223ba10ae494f0d58
3
- size 1549
 
 
 
 
Resources/SafetyChecker.mlmodelc/metadata.json DELETED
@@ -1,130 +0,0 @@
1
- [
2
- {
3
- "shortDescription" : "Stable Diffusion generates images conditioned on text and\/or other images as input through the diffusion process. Please refer to https:\/\/arxiv.org\/abs\/2112.10752 for details.",
4
- "metadataOutputVersion" : "3.0",
5
- "outputSchema" : [
6
- {
7
- "hasShapeFlexibility" : "0",
8
- "isOptional" : "0",
9
- "dataType" : "Float32",
10
- "formattedType" : "MultiArray (Float32 1 × 512 × 512 × 3)",
11
- "shortDescription" : "Identical to the input `images`. If safety checker detected any sensitive content, the corresponding image is replaced with a blank image (zeros)",
12
- "shape" : "[1, 512, 512, 3]",
13
- "name" : "filtered_images",
14
- "type" : "MultiArray"
15
- },
16
- {
17
- "hasShapeFlexibility" : "0",
18
- "isOptional" : "0",
19
- "dataType" : "Float32",
20
- "formattedType" : "MultiArray (Float32 1 × 1 × 1 × 1)",
21
- "shortDescription" : "Indicates whether the safety checker model found any sensitive content in the given image",
22
- "shape" : "[1, 1, 1, 1]",
23
- "name" : "has_nsfw_concepts",
24
- "type" : "MultiArray"
25
- },
26
- {
27
- "hasShapeFlexibility" : "0",
28
- "isOptional" : "0",
29
- "dataType" : "Float32",
30
- "formattedType" : "MultiArray (Float32 1 × 17)",
31
- "shortDescription" : "Concept scores are the scores before thresholding at zero yields the `has_nsfw_concepts` output. These scores can be used to tune the `adjustment` input",
32
- "shape" : "[1, 17]",
33
- "name" : "concept_scores",
34
- "type" : "MultiArray"
35
- }
36
- ],
37
- "version" : "clover-image-tiny-inpaint-v2",
38
- "modelParameters" : [
39
-
40
- ],
41
- "author" : "Please refer to the Model Card available at huggingface.co\/clover-image-tiny-inpaint-v2",
42
- "specificationVersion" : 9,
43
- "storagePrecision" : "Float16",
44
- "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
45
- "mlProgramOperationTypeHistogram" : {
46
- "Ios18.expandDims" : 4,
47
- "Ios18.softmax" : 24,
48
- "Ios18.mul" : 73,
49
- "Ios18.matmul" : 48,
50
- "Ios18.realDiv" : 1,
51
- "Ios18.sigmoid" : 24,
52
- "Ios18.equal" : 1,
53
- "Ios18.greaterEqual" : 1,
54
- "Select" : 2,
55
- "Ios18.greater" : 4,
56
- "Tile" : 4,
57
- "Ios16.reduceSum" : 2,
58
- "Ios18.add" : 52,
59
- "Ios18.layerNorm" : 50,
60
- "Ios18.scatterNd" : 1,
61
- "Ios18.reshape" : 97,
62
- "Shape" : 1,
63
- "Ios18.maximum" : 1,
64
- "Ios18.linear" : 147,
65
- "Ios18.conv" : 1,
66
- "Ios18.concat" : 2,
67
- "Ios18.transpose" : 97,
68
- "Ios18.cast" : 7,
69
- "Ios18.sliceByIndex" : 2,
70
- "NonZero" : 1,
71
- "Ios16.reduceL2Norm" : 1
72
- },
73
- "computePrecision" : "Mixed (Float16, Float32, Int32)",
74
- "stateSchema" : [
75
-
76
- ],
77
- "isUpdatable" : "0",
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
- "inputSchema" : [
90
- {
91
- "hasShapeFlexibility" : "0",
92
- "isOptional" : "0",
93
- "dataType" : "Float16",
94
- "formattedType" : "MultiArray (Float16 1 × 3 × 224 × 224)",
95
- "shortDescription" : "The normalized image input tensor resized to (224x224) in channels-first (BCHW) format",
96
- "shape" : "[1, 3, 224, 224]",
97
- "name" : "clip_input",
98
- "type" : "MultiArray"
99
- },
100
- {
101
- "hasShapeFlexibility" : "0",
102
- "isOptional" : "0",
103
- "dataType" : "Float16",
104
- "formattedType" : "MultiArray (Float16 1 × 512 × 512 × 3)",
105
- "shortDescription" : "Output of the vae_decoder (512x512) in channels-last (BHWC) format",
106
- "shape" : "[1, 512, 512, 3]",
107
- "name" : "images",
108
- "type" : "MultiArray"
109
- },
110
- {
111
- "hasShapeFlexibility" : "0",
112
- "isOptional" : "0",
113
- "dataType" : "Float16",
114
- "formattedType" : "MultiArray (Float16 1)",
115
- "shortDescription" : "Bias added to the concept scores to trade off increased recall for reduce precision in the safety checker classifier",
116
- "shape" : "[1]",
117
- "name" : "adjustment",
118
- "type" : "MultiArray"
119
- }
120
- ],
121
- "userDefinedMetadata" : {
122
- "com.github.apple.coremltools.conversion_date" : "2026-08-11",
123
- "com.github.apple.coremltools.source" : "torch==2.7.0",
124
- "com.github.apple.coremltools.version" : "9.0",
125
- "com.github.apple.coremltools.source_dialect" : "TorchScript"
126
- },
127
- "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_v2_safety_checker",
128
- "method" : "predict"
129
- }
130
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
Resources/SafetyChecker.mlmodelc/model.mil DELETED
The diff for this file is too large to render. See raw diff
 
Resources/SafetyChecker.mlmodelc/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ffdfb1a4774feb816dd2ccc3e304d109013bc9591e188dcab840789583fdc1b4
3
- size 607990114
 
 
 
 
Resources/TextEncoder.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ba6d80c5e94f50f17a70a965c2f63d24fa4239672d0fb5dadcf888e69cafc0f
3
- size 243
 
 
 
 
Resources/TextEncoder.mlmodelc/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:734dc3fdcdcc31367ffae6a58b1e4b0d713eb3f043b89811b7fcaec6df67a495
3
- size 952
 
 
 
 
Resources/TextEncoder.mlmodelc/metadata.json DELETED
@@ -1,90 +0,0 @@
1
- [
2
- {
3
- "shortDescription" : "Stable Diffusion generates images conditioned on text and\/or other images as input through the diffusion process. Please refer to https:\/\/arxiv.org\/abs\/2112.10752 for details.",
4
- "metadataOutputVersion" : "3.0",
5
- "outputSchema" : [
6
- {
7
- "hasShapeFlexibility" : "0",
8
- "isOptional" : "0",
9
- "dataType" : "Float32",
10
- "formattedType" : "MultiArray (Float32 1 × 77 × 768)",
11
- "shortDescription" : "The token embeddings as encoded by the Transformer model",
12
- "shape" : "[1, 77, 768]",
13
- "name" : "last_hidden_state",
14
- "type" : "MultiArray"
15
- },
16
- {
17
- "hasShapeFlexibility" : "0",
18
- "isOptional" : "0",
19
- "dataType" : "Float32",
20
- "formattedType" : "MultiArray (Float32 1 × 768)",
21
- "shortDescription" : "The version of the `last_hidden_state` output after pooling",
22
- "shape" : "[1, 768]",
23
- "name" : "pooled_outputs",
24
- "type" : "MultiArray"
25
- }
26
- ],
27
- "version" : "clover-image-tiny-inpaint-v2",
28
- "modelParameters" : [
29
-
30
- ],
31
- "author" : "Please refer to the Model Card available at huggingface.co\/clover-image-tiny-inpaint-v2",
32
- "specificationVersion" : 9,
33
- "storagePrecision" : "Float16",
34
- "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
35
- "mlProgramOperationTypeHistogram" : {
36
- "Ios18.linear" : 72,
37
- "Ios18.matmul" : 24,
38
- "Select" : 1,
39
- "Ios18.reduceArgmax" : 1,
40
- "Ios18.gather" : 1,
41
- "Ios18.sigmoid" : 12,
42
- "Ios18.add" : 38,
43
- "Stack" : 1,
44
- "Ios18.softmax" : 12,
45
- "Ios18.layerNorm" : 25,
46
- "Ios18.transpose" : 48,
47
- "Ios18.cast" : 4,
48
- "Ios18.gatherNd" : 1,
49
- "Ios18.reshape" : 48,
50
- "Ios18.greaterEqual" : 1,
51
- "Ios18.mul" : 36
52
- },
53
- "computePrecision" : "Mixed (Float16, Float32, Int32, UInt16)",
54
- "stateSchema" : [
55
-
56
- ],
57
- "isUpdatable" : "0",
58
- "availability" : {
59
- "macOS" : "15.0",
60
- "tvOS" : "18.0",
61
- "visionOS" : "2.0",
62
- "watchOS" : "11.0",
63
- "iOS" : "18.0",
64
- "macCatalyst" : "18.0"
65
- },
66
- "modelType" : {
67
- "name" : "MLModelType_mlProgram"
68
- },
69
- "inputSchema" : [
70
- {
71
- "hasShapeFlexibility" : "0",
72
- "isOptional" : "0",
73
- "dataType" : "Float32",
74
- "formattedType" : "MultiArray (Float32 1 × 77)",
75
- "shortDescription" : "The token ids that represent the input text",
76
- "shape" : "[1, 77]",
77
- "name" : "input_ids",
78
- "type" : "MultiArray"
79
- }
80
- ],
81
- "userDefinedMetadata" : {
82
- "com.github.apple.coremltools.conversion_date" : "2026-08-11",
83
- "com.github.apple.coremltools.source" : "torch==2.7.0",
84
- "com.github.apple.coremltools.version" : "9.0",
85
- "com.github.apple.coremltools.source_dialect" : "TorchScript"
86
- },
87
- "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_v2_text_encoder",
88
- "method" : "predict"
89
- }
90
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
Resources/TextEncoder.mlmodelc/model.mil DELETED
The diff for this file is too large to render. See raw diff
 
Resources/TextEncoder.mlmodelc/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:b3e8699e18a386ba82633384e0a6f5c090983f1c66b24a26fad3554fc2a12ddb
3
- size 246145536
 
 
 
 
Resources/UnetChunk1.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:7ee3483e4486c1bdc21299013aa9d3a25fcafb3d064750294c822cf8d6708151
3
- size 243
 
 
 
 
Resources/UnetChunk1.mlmodelc/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:f1ebbe33a5324d71a517efe9d5f5b8f9bbcb9bd2e991a57171840161b3a63430
3
- size 532
 
 
 
 
Resources/UnetChunk1.mlmodelc/metadata.json DELETED
@@ -1,180 +0,0 @@
1
- [
2
- {
3
- "metadataOutputVersion" : "3.0",
4
- "storagePrecision" : "Float16",
5
- "outputSchema" : [
6
- {
7
- "hasShapeFlexibility" : "0",
8
- "isOptional" : "0",
9
- "dataType" : "Float32",
10
- "formattedType" : "MultiArray (Float32 1 × 320 × 64 × 64)",
11
- "shortDescription" : "",
12
- "shape" : "[1, 320, 64, 64]",
13
- "name" : "input_7_cast_fp16",
14
- "type" : "MultiArray"
15
- },
16
- {
17
- "hasShapeFlexibility" : "0",
18
- "isOptional" : "0",
19
- "dataType" : "Float32",
20
- "formattedType" : "MultiArray (Float32 1 × 640 × 32 × 32)",
21
- "shortDescription" : "",
22
- "shape" : "[1, 640, 32, 32]",
23
- "name" : "input_63_cast_fp16",
24
- "type" : "MultiArray"
25
- },
26
- {
27
- "hasShapeFlexibility" : "0",
28
- "isOptional" : "0",
29
- "dataType" : "Float32",
30
- "formattedType" : "MultiArray (Float32 1 × 10240 × 1 × 256)",
31
- "shortDescription" : "",
32
- "shape" : "[1, 10240, 1, 256]",
33
- "name" : "var_2337_cast_fp16",
34
- "type" : "MultiArray"
35
- },
36
- {
37
- "hasShapeFlexibility" : "0",
38
- "isOptional" : "0",
39
- "dataType" : "Float32",
40
- "formattedType" : "MultiArray (Float32 1 × 640 × 16 × 16)",
41
- "shortDescription" : "",
42
- "shape" : "[1, 640, 16, 16]",
43
- "name" : "input_65_cast_fp16",
44
- "type" : "MultiArray"
45
- },
46
- {
47
- "hasShapeFlexibility" : "0",
48
- "isOptional" : "0",
49
- "dataType" : "Float32",
50
- "formattedType" : "MultiArray (Float32 1 × 320 × 32 × 32)",
51
- "shortDescription" : "",
52
- "shape" : "[1, 320, 32, 32]",
53
- "name" : "input_37_cast_fp16",
54
- "type" : "MultiArray"
55
- },
56
- {
57
- "hasShapeFlexibility" : "0",
58
- "isOptional" : "0",
59
- "dataType" : "Float32",
60
- "formattedType" : "MultiArray (Float32 1 × 1280 × 16 × 16)",
61
- "shortDescription" : "",
62
- "shape" : "[1, 1280, 16, 16]",
63
- "name" : "hidden_states_61_cast_fp16",
64
- "type" : "MultiArray"
65
- },
66
- {
67
- "hasShapeFlexibility" : "0",
68
- "isOptional" : "0",
69
- "dataType" : "Float32",
70
- "formattedType" : "MultiArray (Float32 1 × 320 × 64 × 64)",
71
- "shortDescription" : "",
72
- "shape" : "[1, 320, 64, 64]",
73
- "name" : "input_35_cast_fp16",
74
- "type" : "MultiArray"
75
- },
76
- {
77
- "hasShapeFlexibility" : "0",
78
- "isOptional" : "0",
79
- "dataType" : "Float32",
80
- "formattedType" : "MultiArray (Float32 1 × 1280 × 1 × 1)",
81
- "shortDescription" : "",
82
- "shape" : "[1, 1280, 1, 1]",
83
- "name" : "input_15_cast_fp16",
84
- "type" : "MultiArray"
85
- },
86
- {
87
- "hasShapeFlexibility" : "0",
88
- "isOptional" : "0",
89
- "dataType" : "Float32",
90
- "formattedType" : "MultiArray (Float32 1 × 1280 × 1 × 256)",
91
- "shortDescription" : "",
92
- "shape" : "[1, 1280, 1, 256]",
93
- "name" : "inputs_23_cast_fp16",
94
- "type" : "MultiArray"
95
- }
96
- ],
97
- "modelParameters" : [
98
-
99
- ],
100
- "specificationVersion" : 9,
101
- "mlProgramOperationTypeHistogram" : {
102
- "Ios18.expandDims" : 3,
103
- "Ios18.softmax" : 64,
104
- "Ios18.cos" : 1,
105
- "Ios18.mul" : 68,
106
- "Ios18.batchNorm" : 12,
107
- "Ios16.reduceMean" : 24,
108
- "Ios18.realDiv" : 12,
109
- "Ios18.sin" : 1,
110
- "Split" : 3,
111
- "Ios18.add" : 34,
112
- "Ios18.layerNorm" : 12,
113
- "Ios18.reshape" : 31,
114
- "Ios18.conv" : 66,
115
- "Ios18.concat" : 11,
116
- "Ios16.einsum" : 128,
117
- "Ios18.sub" : 12,
118
- "Ios18.silu" : 10,
119
- "Ios18.transpose" : 8,
120
- "Ios18.sqrt" : 12,
121
- "Ios18.gelu" : 3,
122
- "Ios18.square" : 12,
123
- "Ios18.cast" : 9,
124
- "Ios18.sliceByIndex" : 194
125
- },
126
- "computePrecision" : "Mixed (Float16, Float32, Int32)",
127
- "isUpdatable" : "0",
128
- "stateSchema" : [
129
-
130
- ],
131
- "availability" : {
132
- "macOS" : "15.0",
133
- "tvOS" : "18.0",
134
- "visionOS" : "2.0",
135
- "watchOS" : "11.0",
136
- "iOS" : "18.0",
137
- "macCatalyst" : "18.0"
138
- },
139
- "modelType" : {
140
- "name" : "MLModelType_mlProgram"
141
- },
142
- "userDefinedMetadata" : {
143
-
144
- },
145
- "inputSchema" : [
146
- {
147
- "hasShapeFlexibility" : "0",
148
- "isOptional" : "0",
149
- "dataType" : "Float16",
150
- "formattedType" : "MultiArray (Float16 1 × 9 × 64 × 64)",
151
- "shortDescription" : "",
152
- "shape" : "[1, 9, 64, 64]",
153
- "name" : "sample",
154
- "type" : "MultiArray"
155
- },
156
- {
157
- "hasShapeFlexibility" : "0",
158
- "isOptional" : "0",
159
- "dataType" : "Float16",
160
- "formattedType" : "MultiArray (Float16 1)",
161
- "shortDescription" : "",
162
- "shape" : "[1]",
163
- "name" : "timestep",
164
- "type" : "MultiArray"
165
- },
166
- {
167
- "hasShapeFlexibility" : "0",
168
- "isOptional" : "0",
169
- "dataType" : "Float16",
170
- "formattedType" : "MultiArray (Float16 1 × 768 × 1 × 77)",
171
- "shortDescription" : "",
172
- "shape" : "[1, 768, 1, 77]",
173
- "name" : "encoder_hidden_states",
174
- "type" : "MultiArray"
175
- }
176
- ],
177
- "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_v2_unet_chunk1",
178
- "method" : "predict"
179
- }
180
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
Resources/UnetChunk1.mlmodelc/model.mil DELETED
The diff for this file is too large to render. See raw diff
 
Resources/UnetChunk1.mlmodelc/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:091529bc9e2b2bdda0508614e223570f7a8a4fe288bc510ec844635ba2f01046
3
- size 324651584
 
 
 
 
Resources/UnetChunk2.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:1f7d7cba85c9edd629d133c0224baf71e989ad4bfe1da89f37706f1d6cab45ea
3
- size 243
 
 
 
 
Resources/UnetChunk2.mlmodelc/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:94ea7e153dab4f55ad8a40d12f9853422f8bdce240f5b608f3ae72ac3229fb35
3
- size 513
 
 
 
 
Resources/UnetChunk2.mlmodelc/metadata.json DELETED
@@ -1,168 +0,0 @@
1
- [
2
- {
3
- "metadataOutputVersion" : "3.0",
4
- "storagePrecision" : "Float16",
5
- "outputSchema" : [
6
- {
7
- "hasShapeFlexibility" : "0",
8
- "isOptional" : "0",
9
- "dataType" : "Float32",
10
- "formattedType" : "MultiArray (Float32 1 × 4 × 64 × 64)",
11
- "shortDescription" : "",
12
- "shape" : "[1, 4, 64, 64]",
13
- "name" : "noise_pred",
14
- "type" : "MultiArray"
15
- }
16
- ],
17
- "modelParameters" : [
18
-
19
- ],
20
- "specificationVersion" : 9,
21
- "mlProgramOperationTypeHistogram" : {
22
- "Ios18.softmax" : 80,
23
- "Ios18.mul" : 86,
24
- "UpsampleNearestNeighbor" : 2,
25
- "Ios18.batchNorm" : 16,
26
- "Ios16.reduceMean" : 32,
27
- "Ios18.realDiv" : 16,
28
- "Split" : 6,
29
- "Ios18.add" : 48,
30
- "Ios18.layerNorm" : 15,
31
- "Ios18.reshape" : 43,
32
- "Ios18.conv" : 85,
33
- "Ios18.gelu" : 6,
34
- "Ios18.concat" : 15,
35
- "Ios18.sub" : 16,
36
- "Ios16.einsum" : 160,
37
- "Ios18.silu" : 11,
38
- "Ios18.cast" : 10,
39
- "Ios18.transpose" : 10,
40
- "Ios18.square" : 16,
41
- "Ios18.sqrt" : 16,
42
- "Ios18.sliceByIndex" : 240
43
- },
44
- "computePrecision" : "Mixed (Float16, Float32, Int32)",
45
- "isUpdatable" : "0",
46
- "stateSchema" : [
47
-
48
- ],
49
- "availability" : {
50
- "macOS" : "15.0",
51
- "tvOS" : "18.0",
52
- "visionOS" : "2.0",
53
- "watchOS" : "11.0",
54
- "iOS" : "18.0",
55
- "macCatalyst" : "18.0"
56
- },
57
- "modelType" : {
58
- "name" : "MLModelType_mlProgram"
59
- },
60
- "userDefinedMetadata" : {
61
-
62
- },
63
- "inputSchema" : [
64
- {
65
- "hasShapeFlexibility" : "0",
66
- "isOptional" : "0",
67
- "dataType" : "Float16",
68
- "formattedType" : "MultiArray (Float16 1 × 768 × 1 × 77)",
69
- "shortDescription" : "",
70
- "shape" : "[1, 768, 1, 77]",
71
- "name" : "encoder_hidden_states",
72
- "type" : "MultiArray"
73
- },
74
- {
75
- "hasShapeFlexibility" : "0",
76
- "isOptional" : "0",
77
- "dataType" : "Float32",
78
- "formattedType" : "MultiArray (Float32 1 × 640 × 16 × 16)",
79
- "shortDescription" : "",
80
- "shape" : "[1, 640, 16, 16]",
81
- "name" : "input_65_cast_fp16",
82
- "type" : "MultiArray"
83
- },
84
- {
85
- "hasShapeFlexibility" : "0",
86
- "isOptional" : "0",
87
- "dataType" : "Float32",
88
- "formattedType" : "MultiArray (Float32 1 × 1280 × 1 × 256)",
89
- "shortDescription" : "",
90
- "shape" : "[1, 1280, 1, 256]",
91
- "name" : "inputs_23_cast_fp16",
92
- "type" : "MultiArray"
93
- },
94
- {
95
- "hasShapeFlexibility" : "0",
96
- "isOptional" : "0",
97
- "dataType" : "Float32",
98
- "formattedType" : "MultiArray (Float32 1 × 320 × 64 × 64)",
99
- "shortDescription" : "",
100
- "shape" : "[1, 320, 64, 64]",
101
- "name" : "input_35_cast_fp16",
102
- "type" : "MultiArray"
103
- },
104
- {
105
- "hasShapeFlexibility" : "0",
106
- "isOptional" : "0",
107
- "dataType" : "Float32",
108
- "formattedType" : "MultiArray (Float32 1 × 1280 × 16 × 16)",
109
- "shortDescription" : "",
110
- "shape" : "[1, 1280, 16, 16]",
111
- "name" : "hidden_states_61_cast_fp16",
112
- "type" : "MultiArray"
113
- },
114
- {
115
- "hasShapeFlexibility" : "0",
116
- "isOptional" : "0",
117
- "dataType" : "Float32",
118
- "formattedType" : "MultiArray (Float32 1 × 10240 × 1 × 256)",
119
- "shortDescription" : "",
120
- "shape" : "[1, 10240, 1, 256]",
121
- "name" : "var_2337_cast_fp16",
122
- "type" : "MultiArray"
123
- },
124
- {
125
- "hasShapeFlexibility" : "0",
126
- "isOptional" : "0",
127
- "dataType" : "Float32",
128
- "formattedType" : "MultiArray (Float32 1 × 320 × 32 × 32)",
129
- "shortDescription" : "",
130
- "shape" : "[1, 320, 32, 32]",
131
- "name" : "input_37_cast_fp16",
132
- "type" : "MultiArray"
133
- },
134
- {
135
- "hasShapeFlexibility" : "0",
136
- "isOptional" : "0",
137
- "dataType" : "Float32",
138
- "formattedType" : "MultiArray (Float32 1 × 640 × 32 × 32)",
139
- "shortDescription" : "",
140
- "shape" : "[1, 640, 32, 32]",
141
- "name" : "input_63_cast_fp16",
142
- "type" : "MultiArray"
143
- },
144
- {
145
- "hasShapeFlexibility" : "0",
146
- "isOptional" : "0",
147
- "dataType" : "Float32",
148
- "formattedType" : "MultiArray (Float32 1 × 1280 × 1 × 1)",
149
- "shortDescription" : "",
150
- "shape" : "[1, 1280, 1, 1]",
151
- "name" : "input_15_cast_fp16",
152
- "type" : "MultiArray"
153
- },
154
- {
155
- "hasShapeFlexibility" : "0",
156
- "isOptional" : "0",
157
- "dataType" : "Float32",
158
- "formattedType" : "MultiArray (Float32 1 × 320 × 64 × 64)",
159
- "shortDescription" : "",
160
- "shape" : "[1, 320, 64, 64]",
161
- "name" : "input_7_cast_fp16",
162
- "type" : "MultiArray"
163
- }
164
- ],
165
- "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_v2_unet_chunk2",
166
- "method" : "predict"
167
- }
168
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
Resources/UnetChunk2.mlmodelc/model.mil DELETED
The diff for this file is too large to render. See raw diff
 
Resources/UnetChunk2.mlmodelc/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:04910ba6d4ca88b34dd1202c848aebfc33e339010d203573ebc4138a4e9f78ba
3
- size 322211328
 
 
 
 
Resources/VAEDecoder.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ff20b000564aa02c66b7e254258a5c8ba7d34fdb99c54be3c85a42d8eea45a62
3
- size 243
 
 
 
 
Resources/VAEDecoder.mlmodelc/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:7bb1f9ddd844d6c8e14c6e52a1e05d7c493e71091d788798d065141bbe3e131b
3
- size 879
 
 
 
 
Resources/VAEDecoder.mlmodelc/metadata.json DELETED
@@ -1,80 +0,0 @@
1
- [
2
- {
3
- "shortDescription" : "Stable Diffusion generates images conditioned on text and\/or other images as input through the diffusion process. Please refer to https:\/\/arxiv.org\/abs\/2112.10752 for details.",
4
- "metadataOutputVersion" : "3.0",
5
- "outputSchema" : [
6
- {
7
- "hasShapeFlexibility" : "0",
8
- "isOptional" : "0",
9
- "dataType" : "Float32",
10
- "formattedType" : "MultiArray (Float32 1 × 3 × 512 × 512)",
11
- "shortDescription" : "Generated image normalized to range [-1, 1]",
12
- "shape" : "[1, 3, 512, 512]",
13
- "name" : "image",
14
- "type" : "MultiArray"
15
- }
16
- ],
17
- "version" : "clover-image-tiny-inpaint-v2",
18
- "modelParameters" : [
19
-
20
- ],
21
- "author" : "Please refer to the Model Card available at huggingface.co\/clover-image-tiny-inpaint-v2",
22
- "specificationVersion" : 9,
23
- "storagePrecision" : "Float16",
24
- "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
25
- "mlProgramOperationTypeHistogram" : {
26
- "Ios18.square" : 30,
27
- "Ios18.batchNorm" : 29,
28
- "Ios18.conv" : 36,
29
- "Ios18.sub" : 30,
30
- "Ios18.linear" : 4,
31
- "Ios18.scaledDotProductAttention" : 1,
32
- "Ios18.add" : 46,
33
- "Ios18.realDiv" : 30,
34
- "Ios18.silu" : 29,
35
- "Ios16.reduceMean" : 60,
36
- "UpsampleNearestNeighbor" : 3,
37
- "Ios18.transpose" : 6,
38
- "Ios18.cast" : 1,
39
- "Ios18.reshape" : 65,
40
- "Ios18.sqrt" : 30,
41
- "Ios18.mul" : 1
42
- },
43
- "computePrecision" : "Mixed (Float16, Float32, Int32)",
44
- "stateSchema" : [
45
-
46
- ],
47
- "isUpdatable" : "0",
48
- "availability" : {
49
- "macOS" : "15.0",
50
- "tvOS" : "18.0",
51
- "visionOS" : "2.0",
52
- "watchOS" : "11.0",
53
- "iOS" : "18.0",
54
- "macCatalyst" : "18.0"
55
- },
56
- "modelType" : {
57
- "name" : "MLModelType_mlProgram"
58
- },
59
- "inputSchema" : [
60
- {
61
- "hasShapeFlexibility" : "0",
62
- "isOptional" : "0",
63
- "dataType" : "Float16",
64
- "formattedType" : "MultiArray (Float16 1 × 4 × 64 × 64)",
65
- "shortDescription" : "The denoised latent embeddings from the unet model after the last step of reverse diffusion",
66
- "shape" : "[1, 4, 64, 64]",
67
- "name" : "z",
68
- "type" : "MultiArray"
69
- }
70
- ],
71
- "userDefinedMetadata" : {
72
- "com.github.apple.coremltools.conversion_date" : "2026-08-11",
73
- "com.github.apple.coremltools.source" : "torch==2.7.0",
74
- "com.github.apple.coremltools.version" : "9.0",
75
- "com.github.apple.coremltools.source_dialect" : "TorchScript"
76
- },
77
- "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_v2_vae_decoder",
78
- "method" : "predict"
79
- }
80
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
Resources/VAEDecoder.mlmodelc/model.mil DELETED
The diff for this file is too large to render. See raw diff
 
Resources/VAEDecoder.mlmodelc/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ca45a093443d3a9719c5f7c61010315e94ffc8019eff27a0eeb52075a05ac752
3
- size 98993280
 
 
 
 
Resources/VAEEncoder.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ba122c04ae4c8d28c8b8805049e22fb7dfeac3eada7bde68c2366a14e712d1d6
3
- size 243
 
 
 
 
Resources/VAEEncoder.mlmodelc/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:fe1d98a943015c7ac3d29ef2d573b37610e62c97e5724d2f85c555960832629d
3
- size 883
 
 
 
 
Resources/VAEEncoder.mlmodelc/metadata.json DELETED
@@ -1,80 +0,0 @@
1
- [
2
- {
3
- "shortDescription" : "Stable Diffusion generates images conditioned on text and\/or other images as input through the diffusion process. Please refer to https:\/\/arxiv.org\/abs\/2112.10752 for details.",
4
- "metadataOutputVersion" : "3.0",
5
- "outputSchema" : [
6
- {
7
- "hasShapeFlexibility" : "0",
8
- "isOptional" : "0",
9
- "dataType" : "Float32",
10
- "formattedType" : "MultiArray (Float32 1 × 8 × 64 × 64)",
11
- "shortDescription" : "The latent embeddings from the unet model from the input image.",
12
- "shape" : "[1, 8, 64, 64]",
13
- "name" : "latent",
14
- "type" : "MultiArray"
15
- }
16
- ],
17
- "version" : "clover-image-tiny-inpaint-v2",
18
- "modelParameters" : [
19
-
20
- ],
21
- "author" : "Please refer to the Model Card available at huggingface.co\/clover-image-tiny-inpaint-v2",
22
- "specificationVersion" : 9,
23
- "storagePrecision" : "Float16",
24
- "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
25
- "mlProgramOperationTypeHistogram" : {
26
- "Ios18.square" : 22,
27
- "Ios18.batchNorm" : 21,
28
- "Ios18.mul" : 1,
29
- "Ios18.sub" : 22,
30
- "Ios18.conv" : 28,
31
- "Ios18.linear" : 4,
32
- "Ios18.scaledDotProductAttention" : 1,
33
- "Ios18.add" : 34,
34
- "Ios18.realDiv" : 22,
35
- "Ios18.silu" : 21,
36
- "Ios16.reduceMean" : 44,
37
- "Ios18.transpose" : 6,
38
- "Ios18.cast" : 1,
39
- "Ios18.reshape" : 49,
40
- "Ios18.sqrt" : 22,
41
- "Pad" : 3
42
- },
43
- "computePrecision" : "Mixed (Float16, Float32, Int32)",
44
- "stateSchema" : [
45
-
46
- ],
47
- "isUpdatable" : "0",
48
- "availability" : {
49
- "macOS" : "15.0",
50
- "tvOS" : "18.0",
51
- "visionOS" : "2.0",
52
- "watchOS" : "11.0",
53
- "iOS" : "18.0",
54
- "macCatalyst" : "18.0"
55
- },
56
- "modelType" : {
57
- "name" : "MLModelType_mlProgram"
58
- },
59
- "inputSchema" : [
60
- {
61
- "hasShapeFlexibility" : "0",
62
- "isOptional" : "0",
63
- "dataType" : "Float16",
64
- "formattedType" : "MultiArray (Float16 1 × 3 × 512 × 512)",
65
- "shortDescription" : "The input image to base the initial latents on normalized to range [-1, 1]",
66
- "shape" : "[1, 3, 512, 512]",
67
- "name" : "x",
68
- "type" : "MultiArray"
69
- }
70
- ],
71
- "userDefinedMetadata" : {
72
- "com.github.apple.coremltools.conversion_date" : "2026-08-11",
73
- "com.github.apple.coremltools.source" : "torch==2.7.0",
74
- "com.github.apple.coremltools.version" : "9.0",
75
- "com.github.apple.coremltools.source_dialect" : "TorchScript"
76
- },
77
- "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_v2_vae_encoder",
78
- "method" : "predict"
79
- }
80
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
Resources/VAEEncoder.mlmodelc/model.mil DELETED
The diff for this file is too large to render. See raw diff
 
Resources/VAEEncoder.mlmodelc/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:71bf826e5ee1b455462aff0b34e3a469f5f58955ae715708962b1d81db92dc64
3
- size 68338112
 
 
 
 
Resources/merges.txt DELETED
The diff for this file is too large to render. See raw diff
 
Resources/vocab.json DELETED
The diff for this file is too large to render. See raw diff
 
SafetyChecker.mlmodelc/analytics/coremldata.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2ce84e71c14d26c8d655425ebd3299692f1948e213228e9f427f9d7831cc5963
3
  size 243
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8211508a22331e6868fa9d207c6532ae5fc11791e8ceac6bf6fa4c7dd89004e8
3
  size 243
SafetyChecker.mlmodelc/coremldata.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7425b537a536549ab5db59f87b7cde670cfaf1636a64f424710cbeeb70b893b0
3
- size 1543
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f57494e9bc01a089c73ba6b1b9554e1a3df155efbf7cf36223ba10ae494f0d58
3
+ size 1549
SafetyChecker.mlmodelc/metadata.json CHANGED
@@ -34,11 +34,11 @@
34
  "type" : "MultiArray"
35
  }
36
  ],
37
- "version" : "clover-image-tiny-inpaint",
38
  "modelParameters" : [
39
 
40
  ],
41
- "author" : "Please refer to the Model Card available at huggingface.co\/clover-image-tiny-inpaint",
42
  "specificationVersion" : 9,
43
  "storagePrecision" : "Float16",
44
  "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
@@ -119,12 +119,12 @@
119
  }
120
  ],
121
  "userDefinedMetadata" : {
122
- "com.github.apple.coremltools.conversion_date" : "2026-08-10",
123
  "com.github.apple.coremltools.source" : "torch==2.7.0",
124
  "com.github.apple.coremltools.version" : "9.0",
125
  "com.github.apple.coremltools.source_dialect" : "TorchScript"
126
  },
127
- "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_safety_checker",
128
  "method" : "predict"
129
  }
130
  ]
 
34
  "type" : "MultiArray"
35
  }
36
  ],
37
+ "version" : "clover-image-tiny-inpaint-v2",
38
  "modelParameters" : [
39
 
40
  ],
41
+ "author" : "Please refer to the Model Card available at huggingface.co\/clover-image-tiny-inpaint-v2",
42
  "specificationVersion" : 9,
43
  "storagePrecision" : "Float16",
44
  "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
 
119
  }
120
  ],
121
  "userDefinedMetadata" : {
122
+ "com.github.apple.coremltools.conversion_date" : "2026-08-11",
123
  "com.github.apple.coremltools.source" : "torch==2.7.0",
124
  "com.github.apple.coremltools.version" : "9.0",
125
  "com.github.apple.coremltools.source_dialect" : "TorchScript"
126
  },
127
+ "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_v2_safety_checker",
128
  "method" : "predict"
129
  }
130
  ]
TextEncoder.mlmodelc/analytics/coremldata.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:933b2c64d95e0816d52c0b77bd714a0f6e6a2ce7f11219521d8f130ab6adc696
3
  size 243
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ba6d80c5e94f50f17a70a965c2f63d24fa4239672d0fb5dadcf888e69cafc0f
3
  size 243
TextEncoder.mlmodelc/coremldata.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4ce87b6dd6d848c36fefaf1a6808a1bf80839f3eb8d621eb8fdcb6a1077bc5f4
3
- size 946
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:734dc3fdcdcc31367ffae6a58b1e4b0d713eb3f043b89811b7fcaec6df67a495
3
+ size 952
TextEncoder.mlmodelc/metadata.json CHANGED
@@ -24,16 +24,16 @@
24
  "type" : "MultiArray"
25
  }
26
  ],
27
- "version" : "clover-image-tiny-inpaint",
28
  "modelParameters" : [
29
 
30
  ],
31
- "author" : "Please refer to the Model Card available at huggingface.co\/clover-image-tiny-inpaint",
32
  "specificationVersion" : 9,
33
  "storagePrecision" : "Float16",
34
  "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
35
  "mlProgramOperationTypeHistogram" : {
36
- "Ios18.linear" : 71,
37
  "Ios18.matmul" : 24,
38
  "Select" : 1,
39
  "Ios18.reduceArgmax" : 1,
@@ -46,7 +46,7 @@
46
  "Ios18.transpose" : 48,
47
  "Ios18.cast" : 4,
48
  "Ios18.gatherNd" : 1,
49
- "Ios18.reshape" : 47,
50
  "Ios18.greaterEqual" : 1,
51
  "Ios18.mul" : 36
52
  },
@@ -79,12 +79,12 @@
79
  }
80
  ],
81
  "userDefinedMetadata" : {
82
- "com.github.apple.coremltools.conversion_date" : "2026-08-10",
83
  "com.github.apple.coremltools.source" : "torch==2.7.0",
84
  "com.github.apple.coremltools.version" : "9.0",
85
  "com.github.apple.coremltools.source_dialect" : "TorchScript"
86
  },
87
- "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_text_encoder",
88
  "method" : "predict"
89
  }
90
  ]
 
24
  "type" : "MultiArray"
25
  }
26
  ],
27
+ "version" : "clover-image-tiny-inpaint-v2",
28
  "modelParameters" : [
29
 
30
  ],
31
+ "author" : "Please refer to the Model Card available at huggingface.co\/clover-image-tiny-inpaint-v2",
32
  "specificationVersion" : 9,
33
  "storagePrecision" : "Float16",
34
  "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
35
  "mlProgramOperationTypeHistogram" : {
36
+ "Ios18.linear" : 72,
37
  "Ios18.matmul" : 24,
38
  "Select" : 1,
39
  "Ios18.reduceArgmax" : 1,
 
46
  "Ios18.transpose" : 48,
47
  "Ios18.cast" : 4,
48
  "Ios18.gatherNd" : 1,
49
+ "Ios18.reshape" : 48,
50
  "Ios18.greaterEqual" : 1,
51
  "Ios18.mul" : 36
52
  },
 
79
  }
80
  ],
81
  "userDefinedMetadata" : {
82
+ "com.github.apple.coremltools.conversion_date" : "2026-08-11",
83
  "com.github.apple.coremltools.source" : "torch==2.7.0",
84
  "com.github.apple.coremltools.version" : "9.0",
85
  "com.github.apple.coremltools.source_dialect" : "TorchScript"
86
  },
87
+ "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_v2_text_encoder",
88
  "method" : "predict"
89
  }
90
  ]
TextEncoder.mlmodelc/model.mil CHANGED
The diff for this file is too large to render. See raw diff
 
TextEncoder.mlmodelc/weights/weight.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:79b9260c79cbb5b3ed22088fd2765aa5fc1b4b1c24a688537c2439276798ab0c
3
- size 244957824
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3e8699e18a386ba82633384e0a6f5c090983f1c66b24a26fad3554fc2a12ddb
3
+ size 246145536
UnetChunk1.mlmodelc/analytics/coremldata.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1c79e4e6107a0878b51625390eb1389466e2df8b0ca2436f477dc6a2935484c9
3
  size 243
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7ee3483e4486c1bdc21299013aa9d3a25fcafb3d064750294c822cf8d6708151
3
  size 243
UnetChunk1.mlmodelc/coremldata.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4f78341f190d8fd01269c31f7d0a7576f3df9cafa007f7dcf47915d9d598d4da
3
  size 532
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1ebbe33a5324d71a517efe9d5f5b8f9bbcb9bd2e991a57171840161b3a63430
3
  size 532
UnetChunk1.mlmodelc/metadata.json CHANGED
@@ -7,20 +7,20 @@
7
  "hasShapeFlexibility" : "0",
8
  "isOptional" : "0",
9
  "dataType" : "Float32",
10
- "formattedType" : "MultiArray (Float32 1 × 640 × 16 × 16)",
11
  "shortDescription" : "",
12
- "shape" : "[1, 640, 16, 16]",
13
- "name" : "input_65_cast_fp16",
14
  "type" : "MultiArray"
15
  },
16
  {
17
  "hasShapeFlexibility" : "0",
18
  "isOptional" : "0",
19
  "dataType" : "Float32",
20
- "formattedType" : "MultiArray (Float32 1 × 1280 × 1 × 256)",
21
  "shortDescription" : "",
22
- "shape" : "[1, 1280, 1, 256]",
23
- "name" : "inputs_23_cast_fp16",
24
  "type" : "MultiArray"
25
  },
26
  {
@@ -37,20 +37,20 @@
37
  "hasShapeFlexibility" : "0",
38
  "isOptional" : "0",
39
  "dataType" : "Float32",
40
- "formattedType" : "MultiArray (Float32 1 × 1280 × 1 × 1)",
41
  "shortDescription" : "",
42
- "shape" : "[1, 1280, 1, 1]",
43
- "name" : "input_15_cast_fp16",
44
  "type" : "MultiArray"
45
  },
46
  {
47
  "hasShapeFlexibility" : "0",
48
  "isOptional" : "0",
49
  "dataType" : "Float32",
50
- "formattedType" : "MultiArray (Float32 1 × 320 × 64 × 64)",
51
  "shortDescription" : "",
52
- "shape" : "[1, 320, 64, 64]",
53
- "name" : "input_7_cast_fp16",
54
  "type" : "MultiArray"
55
  },
56
  {
@@ -77,20 +77,20 @@
77
  "hasShapeFlexibility" : "0",
78
  "isOptional" : "0",
79
  "dataType" : "Float32",
80
- "formattedType" : "MultiArray (Float32 1 × 640 × 32 × 32)",
81
  "shortDescription" : "",
82
- "shape" : "[1, 640, 32, 32]",
83
- "name" : "input_63_cast_fp16",
84
  "type" : "MultiArray"
85
  },
86
  {
87
  "hasShapeFlexibility" : "0",
88
  "isOptional" : "0",
89
  "dataType" : "Float32",
90
- "formattedType" : "MultiArray (Float32 1 × 320 × 32 × 32)",
91
  "shortDescription" : "",
92
- "shape" : "[1, 320, 32, 32]",
93
- "name" : "input_37_cast_fp16",
94
  "type" : "MultiArray"
95
  }
96
  ],
@@ -174,7 +174,7 @@
174
  "type" : "MultiArray"
175
  }
176
  ],
177
- "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_unet_chunk1",
178
  "method" : "predict"
179
  }
180
  ]
 
7
  "hasShapeFlexibility" : "0",
8
  "isOptional" : "0",
9
  "dataType" : "Float32",
10
+ "formattedType" : "MultiArray (Float32 1 × 320 × 64 × 64)",
11
  "shortDescription" : "",
12
+ "shape" : "[1, 320, 64, 64]",
13
+ "name" : "input_7_cast_fp16",
14
  "type" : "MultiArray"
15
  },
16
  {
17
  "hasShapeFlexibility" : "0",
18
  "isOptional" : "0",
19
  "dataType" : "Float32",
20
+ "formattedType" : "MultiArray (Float32 1 × 640 × 32 × 32)",
21
  "shortDescription" : "",
22
+ "shape" : "[1, 640, 32, 32]",
23
+ "name" : "input_63_cast_fp16",
24
  "type" : "MultiArray"
25
  },
26
  {
 
37
  "hasShapeFlexibility" : "0",
38
  "isOptional" : "0",
39
  "dataType" : "Float32",
40
+ "formattedType" : "MultiArray (Float32 1 × 640 × 16 × 16)",
41
  "shortDescription" : "",
42
+ "shape" : "[1, 640, 16, 16]",
43
+ "name" : "input_65_cast_fp16",
44
  "type" : "MultiArray"
45
  },
46
  {
47
  "hasShapeFlexibility" : "0",
48
  "isOptional" : "0",
49
  "dataType" : "Float32",
50
+ "formattedType" : "MultiArray (Float32 1 × 320 × 32 × 32)",
51
  "shortDescription" : "",
52
+ "shape" : "[1, 320, 32, 32]",
53
+ "name" : "input_37_cast_fp16",
54
  "type" : "MultiArray"
55
  },
56
  {
 
77
  "hasShapeFlexibility" : "0",
78
  "isOptional" : "0",
79
  "dataType" : "Float32",
80
+ "formattedType" : "MultiArray (Float32 1 × 1280 × 1 × 1)",
81
  "shortDescription" : "",
82
+ "shape" : "[1, 1280, 1, 1]",
83
+ "name" : "input_15_cast_fp16",
84
  "type" : "MultiArray"
85
  },
86
  {
87
  "hasShapeFlexibility" : "0",
88
  "isOptional" : "0",
89
  "dataType" : "Float32",
90
+ "formattedType" : "MultiArray (Float32 1 × 1280 × 1 × 256)",
91
  "shortDescription" : "",
92
+ "shape" : "[1, 1280, 1, 256]",
93
+ "name" : "inputs_23_cast_fp16",
94
  "type" : "MultiArray"
95
  }
96
  ],
 
174
  "type" : "MultiArray"
175
  }
176
  ],
177
+ "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_v2_unet_chunk1",
178
  "method" : "predict"
179
  }
180
  ]
UnetChunk1.mlmodelc/model.mil CHANGED
@@ -2155,23 +2155,23 @@ program(1.3)
2155
  tensor<fp16, [10240, 1280, 1, 1]> up_blocks_0_attentions_0_transformer_blocks_0_ff_net_0_proj_weight_to_fp16 = const()[name = string("up_blocks_0_attentions_0_transformer_blocks_0_ff_net_0_proj_weight_to_fp16"), val = tensor<fp16, [10240, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(298416576)))];
2156
  tensor<fp16, [10240]> up_blocks_0_attentions_0_transformer_blocks_0_ff_net_0_proj_bias_to_fp16 = const()[name = string("up_blocks_0_attentions_0_transformer_blocks_0_ff_net_0_proj_bias_to_fp16"), val = tensor<fp16, [10240]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(324631040)))];
2157
  tensor<fp16, [1, 10240, 1, 256]> var_2337_cast_fp16_1 = conv(bias = up_blocks_0_attentions_0_transformer_blocks_0_ff_net_0_proj_bias_to_fp16, dilations = var_2337_dilations_0, groups = var_2337_groups_0, pad = var_2337_pad_0, pad_type = var_2337_pad_type_0, strides = var_2337_strides_0, weight = up_blocks_0_attentions_0_transformer_blocks_0_ff_net_0_proj_weight_to_fp16, x = input_111_cast_fp16)[name = string("op_2337_cast_fp16")];
2158
- string input_65_cast_fp16_dtype_0 = const()[name = string("input_65_cast_fp16_dtype_0"), val = string("fp32")];
2159
- string inputs_23_cast_fp16_dtype_0 = const()[name = string("inputs_23_cast_fp16_dtype_0"), val = string("fp32")];
2160
- string var_2337_cast_fp16_dtype_0 = const()[name = string("var_2337_cast_fp16_dtype_0"), val = string("fp32")];
2161
- string input_15_cast_fp16_dtype_0 = const()[name = string("input_15_cast_fp16_dtype_0"), val = string("fp32")];
2162
  string input_7_cast_fp16_dtype_0 = const()[name = string("input_7_cast_fp16_dtype_0"), val = string("fp32")];
2163
- string hidden_states_61_cast_fp16_dtype_0 = const()[name = string("hidden_states_61_cast_fp16_dtype_0"), val = string("fp32")];
2164
- string input_35_cast_fp16_dtype_0 = const()[name = string("input_35_cast_fp16_dtype_0"), val = string("fp32")];
2165
  string input_63_cast_fp16_dtype_0 = const()[name = string("input_63_cast_fp16_dtype_0"), val = string("fp32")];
 
 
2166
  string input_37_cast_fp16_dtype_0 = const()[name = string("input_37_cast_fp16_dtype_0"), val = string("fp32")];
2167
- tensor<fp32, [1, 320, 32, 32]> input_37_cast_fp16 = cast(dtype = input_37_cast_fp16_dtype_0, x = input_37_cast_fp16_1)[name = string("cast_9")];
2168
- tensor<fp32, [1, 640, 32, 32]> input_63_cast_fp16 = cast(dtype = input_63_cast_fp16_dtype_0, x = input_63_cast_fp16_1)[name = string("cast_10")];
 
 
 
 
2169
  tensor<fp32, [1, 320, 64, 64]> input_35_cast_fp16 = cast(dtype = input_35_cast_fp16_dtype_0, x = input_35_cast_fp16_1)[name = string("cast_11")];
2170
  tensor<fp32, [1, 1280, 16, 16]> hidden_states_61_cast_fp16 = cast(dtype = hidden_states_61_cast_fp16_dtype_0, x = hidden_states_61_cast_fp16_1)[name = string("cast_12")];
2171
- tensor<fp32, [1, 320, 64, 64]> input_7_cast_fp16 = cast(dtype = input_7_cast_fp16_dtype_0, x = input_7_cast_fp16_1)[name = string("cast_13")];
2172
- tensor<fp32, [1, 1280, 1, 1]> input_15_cast_fp16 = cast(dtype = input_15_cast_fp16_dtype_0, x = input_15_cast_fp16_1)[name = string("cast_14")];
2173
  tensor<fp32, [1, 10240, 1, 256]> var_2337_cast_fp16 = cast(dtype = var_2337_cast_fp16_dtype_0, x = var_2337_cast_fp16_1)[name = string("cast_15")];
2174
- tensor<fp32, [1, 1280, 1, 256]> inputs_23_cast_fp16 = cast(dtype = inputs_23_cast_fp16_dtype_0, x = inputs_23_cast_fp16_1)[name = string("cast_16")];
2175
- tensor<fp32, [1, 640, 16, 16]> input_65_cast_fp16 = cast(dtype = input_65_cast_fp16_dtype_0, x = input_65_cast_fp16_1)[name = string("cast_17")];
2176
- } -> (input_65_cast_fp16, inputs_23_cast_fp16, var_2337_cast_fp16, input_15_cast_fp16, input_7_cast_fp16, hidden_states_61_cast_fp16, input_35_cast_fp16, input_63_cast_fp16, input_37_cast_fp16);
2177
  }
 
2155
  tensor<fp16, [10240, 1280, 1, 1]> up_blocks_0_attentions_0_transformer_blocks_0_ff_net_0_proj_weight_to_fp16 = const()[name = string("up_blocks_0_attentions_0_transformer_blocks_0_ff_net_0_proj_weight_to_fp16"), val = tensor<fp16, [10240, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(298416576)))];
2156
  tensor<fp16, [10240]> up_blocks_0_attentions_0_transformer_blocks_0_ff_net_0_proj_bias_to_fp16 = const()[name = string("up_blocks_0_attentions_0_transformer_blocks_0_ff_net_0_proj_bias_to_fp16"), val = tensor<fp16, [10240]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(324631040)))];
2157
  tensor<fp16, [1, 10240, 1, 256]> var_2337_cast_fp16_1 = conv(bias = up_blocks_0_attentions_0_transformer_blocks_0_ff_net_0_proj_bias_to_fp16, dilations = var_2337_dilations_0, groups = var_2337_groups_0, pad = var_2337_pad_0, pad_type = var_2337_pad_type_0, strides = var_2337_strides_0, weight = up_blocks_0_attentions_0_transformer_blocks_0_ff_net_0_proj_weight_to_fp16, x = input_111_cast_fp16)[name = string("op_2337_cast_fp16")];
 
 
 
 
2158
  string input_7_cast_fp16_dtype_0 = const()[name = string("input_7_cast_fp16_dtype_0"), val = string("fp32")];
 
 
2159
  string input_63_cast_fp16_dtype_0 = const()[name = string("input_63_cast_fp16_dtype_0"), val = string("fp32")];
2160
+ string var_2337_cast_fp16_dtype_0 = const()[name = string("var_2337_cast_fp16_dtype_0"), val = string("fp32")];
2161
+ string input_65_cast_fp16_dtype_0 = const()[name = string("input_65_cast_fp16_dtype_0"), val = string("fp32")];
2162
  string input_37_cast_fp16_dtype_0 = const()[name = string("input_37_cast_fp16_dtype_0"), val = string("fp32")];
2163
+ string hidden_states_61_cast_fp16_dtype_0 = const()[name = string("hidden_states_61_cast_fp16_dtype_0"), val = string("fp32")];
2164
+ string input_35_cast_fp16_dtype_0 = const()[name = string("input_35_cast_fp16_dtype_0"), val = string("fp32")];
2165
+ string input_15_cast_fp16_dtype_0 = const()[name = string("input_15_cast_fp16_dtype_0"), val = string("fp32")];
2166
+ string inputs_23_cast_fp16_dtype_0 = const()[name = string("inputs_23_cast_fp16_dtype_0"), val = string("fp32")];
2167
+ tensor<fp32, [1, 1280, 1, 256]> inputs_23_cast_fp16 = cast(dtype = inputs_23_cast_fp16_dtype_0, x = inputs_23_cast_fp16_1)[name = string("cast_9")];
2168
+ tensor<fp32, [1, 1280, 1, 1]> input_15_cast_fp16 = cast(dtype = input_15_cast_fp16_dtype_0, x = input_15_cast_fp16_1)[name = string("cast_10")];
2169
  tensor<fp32, [1, 320, 64, 64]> input_35_cast_fp16 = cast(dtype = input_35_cast_fp16_dtype_0, x = input_35_cast_fp16_1)[name = string("cast_11")];
2170
  tensor<fp32, [1, 1280, 16, 16]> hidden_states_61_cast_fp16 = cast(dtype = hidden_states_61_cast_fp16_dtype_0, x = hidden_states_61_cast_fp16_1)[name = string("cast_12")];
2171
+ tensor<fp32, [1, 320, 32, 32]> input_37_cast_fp16 = cast(dtype = input_37_cast_fp16_dtype_0, x = input_37_cast_fp16_1)[name = string("cast_13")];
2172
+ tensor<fp32, [1, 640, 16, 16]> input_65_cast_fp16 = cast(dtype = input_65_cast_fp16_dtype_0, x = input_65_cast_fp16_1)[name = string("cast_14")];
2173
  tensor<fp32, [1, 10240, 1, 256]> var_2337_cast_fp16 = cast(dtype = var_2337_cast_fp16_dtype_0, x = var_2337_cast_fp16_1)[name = string("cast_15")];
2174
+ tensor<fp32, [1, 640, 32, 32]> input_63_cast_fp16 = cast(dtype = input_63_cast_fp16_dtype_0, x = input_63_cast_fp16_1)[name = string("cast_16")];
2175
+ tensor<fp32, [1, 320, 64, 64]> input_7_cast_fp16 = cast(dtype = input_7_cast_fp16_dtype_0, x = input_7_cast_fp16_1)[name = string("cast_17")];
2176
+ } -> (input_7_cast_fp16, input_63_cast_fp16, var_2337_cast_fp16, input_65_cast_fp16, input_37_cast_fp16, hidden_states_61_cast_fp16, input_35_cast_fp16, input_15_cast_fp16, inputs_23_cast_fp16);
2177
  }
UnetChunk1.mlmodelc/weights/weight.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:15d72e32d29f8aa5813a0bc9fca0ccdf1063411b91c496ec20fa6a25af8fdc13
3
  size 324651584
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:091529bc9e2b2bdda0508614e223570f7a8a4fe288bc510ec844635ba2f01046
3
  size 324651584
UnetChunk2.mlmodelc/analytics/coremldata.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3a468503aee82948546a734364424194ba66fe01ca5d4fd403fc75f7569234ed
3
  size 243
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f7d7cba85c9edd629d133c0224baf71e989ad4bfe1da89f37706f1d6cab45ea
3
  size 243
UnetChunk2.mlmodelc/coremldata.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:45154abdc0829837df1817882dc4d6fe759241b60a55a316f735785999e3daac
3
  size 513
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:94ea7e153dab4f55ad8a40d12f9853422f8bdce240f5b608f3ae72ac3229fb35
3
  size 513
UnetChunk2.mlmodelc/metadata.json CHANGED
@@ -71,16 +71,6 @@
71
  "name" : "encoder_hidden_states",
72
  "type" : "MultiArray"
73
  },
74
- {
75
- "hasShapeFlexibility" : "0",
76
- "isOptional" : "0",
77
- "dataType" : "Float32",
78
- "formattedType" : "MultiArray (Float32 1 × 320 × 64 × 64)",
79
- "shortDescription" : "",
80
- "shape" : "[1, 320, 64, 64]",
81
- "name" : "input_7_cast_fp16",
82
- "type" : "MultiArray"
83
- },
84
  {
85
  "hasShapeFlexibility" : "0",
86
  "isOptional" : "0",
@@ -105,10 +95,10 @@
105
  "hasShapeFlexibility" : "0",
106
  "isOptional" : "0",
107
  "dataType" : "Float32",
108
- "formattedType" : "MultiArray (Float32 1 × 1280 × 1 × 1)",
109
  "shortDescription" : "",
110
- "shape" : "[1, 1280, 1, 1]",
111
- "name" : "input_15_cast_fp16",
112
  "type" : "MultiArray"
113
  },
114
  {
@@ -121,16 +111,6 @@
121
  "name" : "hidden_states_61_cast_fp16",
122
  "type" : "MultiArray"
123
  },
124
- {
125
- "hasShapeFlexibility" : "0",
126
- "isOptional" : "0",
127
- "dataType" : "Float32",
128
- "formattedType" : "MultiArray (Float32 1 × 320 × 64 × 64)",
129
- "shortDescription" : "",
130
- "shape" : "[1, 320, 64, 64]",
131
- "name" : "input_35_cast_fp16",
132
- "type" : "MultiArray"
133
- },
134
  {
135
  "hasShapeFlexibility" : "0",
136
  "isOptional" : "0",
@@ -160,9 +140,29 @@
160
  "shape" : "[1, 640, 32, 32]",
161
  "name" : "input_63_cast_fp16",
162
  "type" : "MultiArray"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
163
  }
164
  ],
165
- "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_unet_chunk2",
166
  "method" : "predict"
167
  }
168
  ]
 
71
  "name" : "encoder_hidden_states",
72
  "type" : "MultiArray"
73
  },
 
 
 
 
 
 
 
 
 
 
74
  {
75
  "hasShapeFlexibility" : "0",
76
  "isOptional" : "0",
 
95
  "hasShapeFlexibility" : "0",
96
  "isOptional" : "0",
97
  "dataType" : "Float32",
98
+ "formattedType" : "MultiArray (Float32 1 × 320 × 64 × 64)",
99
  "shortDescription" : "",
100
+ "shape" : "[1, 320, 64, 64]",
101
+ "name" : "input_35_cast_fp16",
102
  "type" : "MultiArray"
103
  },
104
  {
 
111
  "name" : "hidden_states_61_cast_fp16",
112
  "type" : "MultiArray"
113
  },
 
 
 
 
 
 
 
 
 
 
114
  {
115
  "hasShapeFlexibility" : "0",
116
  "isOptional" : "0",
 
140
  "shape" : "[1, 640, 32, 32]",
141
  "name" : "input_63_cast_fp16",
142
  "type" : "MultiArray"
143
+ },
144
+ {
145
+ "hasShapeFlexibility" : "0",
146
+ "isOptional" : "0",
147
+ "dataType" : "Float32",
148
+ "formattedType" : "MultiArray (Float32 1 × 1280 × 1 × 1)",
149
+ "shortDescription" : "",
150
+ "shape" : "[1, 1280, 1, 1]",
151
+ "name" : "input_15_cast_fp16",
152
+ "type" : "MultiArray"
153
+ },
154
+ {
155
+ "hasShapeFlexibility" : "0",
156
+ "isOptional" : "0",
157
+ "dataType" : "Float32",
158
+ "formattedType" : "MultiArray (Float32 1 × 320 × 64 × 64)",
159
+ "shortDescription" : "",
160
+ "shape" : "[1, 320, 64, 64]",
161
+ "name" : "input_7_cast_fp16",
162
+ "type" : "MultiArray"
163
  }
164
  ],
165
+ "generatedClassName" : "Stable_Diffusion_version_clover_image_tiny_inpaint_v2_unet_chunk2",
166
  "method" : "predict"
167
  }
168
  ]
UnetChunk2.mlmodelc/model.mil CHANGED
@@ -2,26 +2,26 @@ program(1.3)
2
  [buildInfo = dict<string, string>({{"coremlc-component-MIL", "3600.16.1"}, {"coremlc-version", "3600.22.1"}})]
3
  {
4
  func main<ios18>(tensor<fp16, [1, 768, 1, 77]> encoder_hidden_states, tensor<fp32, [1, 1280, 16, 16]> hidden_states_61_cast_fp16, tensor<fp32, [1, 1280, 1, 1]> input_15_cast_fp16, tensor<fp32, [1, 320, 64, 64]> input_35_cast_fp16, tensor<fp32, [1, 320, 32, 32]> input_37_cast_fp16, tensor<fp32, [1, 640, 32, 32]> input_63_cast_fp16, tensor<fp32, [1, 640, 16, 16]> input_65_cast_fp16, tensor<fp32, [1, 320, 64, 64]> input_7_cast_fp16, tensor<fp32, [1, 1280, 1, 256]> inputs_23_cast_fp16, tensor<fp32, [1, 10240, 1, 256]> var_2337_cast_fp16) {
5
- string cast_0_dtype_0 = const()[name = string("cast_0_dtype_0"), val = string("fp16")];
6
  tensor<fp16, [320]> add_1_mean_0_to_fp16 = const()[name = string("add_1_mean_0_to_fp16"), val = tensor<fp16, [320]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
7
  tensor<fp16, [320]> add_1_variance_0_to_fp16 = const()[name = string("add_1_variance_0_to_fp16"), val = tensor<fp16, [320]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(768)))];
8
- string cast_3_dtype_0 = const()[name = string("cast_3_dtype_0"), val = string("fp16")];
9
- string cast_5_dtype_0 = const()[name = string("cast_5_dtype_0"), val = string("fp16")];
10
  string cast_7_dtype_0 = const()[name = string("cast_7_dtype_0"), val = string("fp16")];
 
 
11
  tensor<fp16, [640]> add_9_mean_0_to_fp16 = const()[name = string("add_9_mean_0_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1472)))];
12
  tensor<fp16, [640]> add_9_variance_0_to_fp16 = const()[name = string("add_9_variance_0_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2816)))];
13
- string cast_8_dtype_0 = const()[name = string("cast_8_dtype_0"), val = string("fp16")];
14
- string cast_1_dtype_0 = const()[name = string("cast_1_dtype_0"), val = string("fp16")];
15
  tensor<fp16, [1280]> add_15_mean_0_to_fp16 = const()[name = string("add_15_mean_0_to_fp16"), val = tensor<fp16, [1280]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4160)))];
16
  tensor<fp16, [1280]> add_15_variance_0_to_fp16 = const()[name = string("add_15_variance_0_to_fp16"), val = tensor<fp16, [1280]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6784)))];
17
  int32 var_1812 = const()[name = string("op_1812"), val = int32(1)];
 
 
18
  string cast_4_dtype_0 = const()[name = string("cast_4_dtype_0"), val = string("fp16")];
19
- string cast_2_dtype_0 = const()[name = string("cast_2_dtype_0"), val = string("fp16")];
20
- string cast_6_dtype_0 = const()[name = string("cast_6_dtype_0"), val = string("fp16")];
21
  tensor<int32, [2]> var_2338_split_sizes_0 = const()[name = string("op_2338_split_sizes_0"), val = tensor<int32, [2]>([5120, 5120])];
22
  int32 var_2338_axis_0 = const()[name = string("op_2338_axis_0"), val = int32(1)];
23
- tensor<fp16, [1, 10240, 1, 256]> cast_6 = cast(dtype = cast_6_dtype_0, x = var_2337_cast_fp16)[name = string("cast_1")];
24
- tensor<fp16, [1, 5120, 1, 256]> var_2338_cast_fp16_0, tensor<fp16, [1, 5120, 1, 256]> var_2338_cast_fp16_1 = split(axis = var_2338_axis_0, split_sizes = var_2338_split_sizes_0, x = cast_6)[name = string("op_2338_cast_fp16")];
25
  string var_2340_mode_0 = const()[name = string("op_2340_mode_0"), val = string("EXACT")];
26
  tensor<fp16, [1, 5120, 1, 256]> var_2340_cast_fp16 = gelu(mode = var_2340_mode_0, x = var_2338_cast_fp16_1)[name = string("op_2340_cast_fp16")];
27
  tensor<fp16, [1, 5120, 1, 256]> input_113_cast_fp16 = mul(x = var_2338_cast_fp16_0, y = var_2340_cast_fp16)[name = string("input_113_cast_fp16")];
@@ -33,8 +33,8 @@ program(1.3)
33
  tensor<fp16, [1280, 5120, 1, 1]> up_blocks_0_attentions_0_transformer_blocks_0_ff_net_2_weight_to_fp16 = const()[name = string("up_blocks_0_attentions_0_transformer_blocks_0_ff_net_2_weight_to_fp16"), val = tensor<fp16, [1280, 5120, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(9408)))];
34
  tensor<fp16, [1280]> up_blocks_0_attentions_0_transformer_blocks_0_ff_net_2_bias_to_fp16 = const()[name = string("up_blocks_0_attentions_0_transformer_blocks_0_ff_net_2_bias_to_fp16"), val = tensor<fp16, [1280]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13116672)))];
35
  tensor<fp16, [1, 1280, 1, 256]> var_2348_cast_fp16 = conv(bias = up_blocks_0_attentions_0_transformer_blocks_0_ff_net_2_bias_to_fp16, dilations = var_2348_dilations_0, groups = var_2348_groups_0, pad = var_2348_pad_0, pad_type = var_2348_pad_type_0, strides = var_2348_strides_0, weight = up_blocks_0_attentions_0_transformer_blocks_0_ff_net_2_weight_to_fp16, x = input_113_cast_fp16)[name = string("op_2348_cast_fp16")];
36
- tensor<fp16, [1, 1280, 1, 256]> cast_2 = cast(dtype = cast_2_dtype_0, x = inputs_23_cast_fp16)[name = string("cast_2")];
37
- tensor<fp16, [1, 1280, 1, 256]> hidden_states_71_cast_fp16 = add(x = var_2348_cast_fp16, y = cast_2)[name = string("hidden_states_71_cast_fp16")];
38
  tensor<int32, [4]> var_2350 = const()[name = string("op_2350"), val = tensor<int32, [4]>([1, 1280, 16, 16])];
39
  tensor<fp16, [1, 1280, 16, 16]> input_115_cast_fp16 = reshape(shape = var_2350, x = hidden_states_71_cast_fp16)[name = string("input_115_cast_fp16")];
40
  string hidden_states_73_pad_type_0 = const()[name = string("hidden_states_73_pad_type_0"), val = string("valid")];
@@ -45,11 +45,11 @@ program(1.3)
45
  tensor<fp16, [1280, 1280, 1, 1]> up_blocks_0_attentions_0_proj_out_weight_to_fp16 = const()[name = string("up_blocks_0_attentions_0_proj_out_weight_to_fp16"), val = tensor<fp16, [1280, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13119296)))];
46
  tensor<fp16, [1280]> up_blocks_0_attentions_0_proj_out_bias_to_fp16 = const()[name = string("up_blocks_0_attentions_0_proj_out_bias_to_fp16"), val = tensor<fp16, [1280]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(16396160)))];
47
  tensor<fp16, [1, 1280, 16, 16]> hidden_states_73_cast_fp16 = conv(bias = up_blocks_0_attentions_0_proj_out_bias_to_fp16, dilations = hidden_states_73_dilations_0, groups = hidden_states_73_groups_0, pad = hidden_states_73_pad_0, pad_type = hidden_states_73_pad_type_0, strides = hidden_states_73_strides_0, weight = up_blocks_0_attentions_0_proj_out_weight_to_fp16, x = input_115_cast_fp16)[name = string("hidden_states_73_cast_fp16")];
48
- tensor<fp16, [1, 1280, 16, 16]> cast_4 = cast(dtype = cast_4_dtype_0, x = hidden_states_61_cast_fp16)[name = string("cast_3")];
49
- tensor<fp16, [1, 1280, 16, 16]> hidden_states_75_cast_fp16 = add(x = hidden_states_73_cast_fp16, y = cast_4)[name = string("hidden_states_75_cast_fp16")];
50
  bool input_117_interleave_0 = const()[name = string("input_117_interleave_0"), val = bool(false)];
51
- tensor<fp16, [1, 640, 16, 16]> cast_1 = cast(dtype = cast_1_dtype_0, x = input_65_cast_fp16)[name = string("cast_4")];
52
- tensor<fp16, [1, 1920, 16, 16]> input_117_cast_fp16 = concat(axis = var_1812, interleave = input_117_interleave_0, values = (hidden_states_75_cast_fp16, cast_1))[name = string("input_117_cast_fp16")];
53
  tensor<int32, [5]> reshape_48_shape_0 = const()[name = string("reshape_48_shape_0"), val = tensor<int32, [5]>([1, 32, 60, 16, 16])];
54
  tensor<fp16, [1, 32, 60, 16, 16]> reshape_48_cast_fp16 = reshape(shape = reshape_48_shape_0, x = input_117_cast_fp16)[name = string("reshape_48_cast_fp16")];
55
  tensor<int32, [3]> reduce_mean_36_axes_0 = const()[name = string("reduce_mean_36_axes_0"), val = tensor<int32, [3]>([2, 3, 4])];
@@ -88,8 +88,8 @@ program(1.3)
88
  int32 temb_9_groups_0 = const()[name = string("temb_9_groups_0"), val = int32(1)];
89
  tensor<fp16, [1280, 1280, 1, 1]> up_blocks_0_resnets_1_time_emb_proj_weight_to_fp16 = const()[name = string("up_blocks_0_resnets_1_time_emb_proj_weight_to_fp16"), val = tensor<fp16, [1280, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(60653888)))];
90
  tensor<fp16, [1280]> up_blocks_0_resnets_1_time_emb_proj_bias_to_fp16 = const()[name = string("up_blocks_0_resnets_1_time_emb_proj_bias_to_fp16"), val = tensor<fp16, [1280]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(63930752)))];
91
- tensor<fp16, [1, 1280, 1, 1]> cast_3 = cast(dtype = cast_3_dtype_0, x = input_15_cast_fp16)[name = string("cast_8")];
92
- tensor<fp16, [1, 1280, 1, 1]> temb_9_cast_fp16 = conv(bias = up_blocks_0_resnets_1_time_emb_proj_bias_to_fp16, dilations = temb_9_dilations_0, groups = temb_9_groups_0, pad = temb_9_pad_0, pad_type = temb_9_pad_type_0, strides = temb_9_strides_0, weight = up_blocks_0_resnets_1_time_emb_proj_weight_to_fp16, x = cast_3)[name = string("temb_9_cast_fp16")];
93
  tensor<fp16, [1, 1280, 16, 16]> input_125_cast_fp16 = add(x = hidden_states_77_cast_fp16, y = temb_9_cast_fp16)[name = string("input_125_cast_fp16")];
94
  tensor<int32, [5]> reshape_52_shape_0 = const()[name = string("reshape_52_shape_0"), val = tensor<int32, [5]>([1, 32, 40, 16, 16])];
95
  tensor<fp16, [1, 32, 40, 16, 16]> reshape_52_cast_fp16 = reshape(shape = reshape_52_shape_0, x = input_125_cast_fp16)[name = string("reshape_52_cast_fp16")];
@@ -593,8 +593,8 @@ program(1.3)
593
  tensor<fp16, [1, 1280, 32, 32]> hidden_states_95_cast_fp16 = conv(bias = up_blocks_0_upsamplers_0_conv_bias_to_fp16, dilations = hidden_states_95_dilations_0, groups = hidden_states_95_groups_0, pad = hidden_states_95_pad_0, pad_type = hidden_states_95_pad_type_0, strides = hidden_states_95_strides_0, weight = up_blocks_0_upsamplers_0_conv_weight_to_fp16, x = input_145_cast_fp16)[name = string("hidden_states_95_cast_fp16")];
594
  int32 var_2929 = const()[name = string("op_2929"), val = int32(1)];
595
  bool input_147_interleave_0 = const()[name = string("input_147_interleave_0"), val = bool(false)];
596
- tensor<fp16, [1, 640, 32, 32]> cast_8 = cast(dtype = cast_8_dtype_0, x = input_63_cast_fp16)[name = string("cast_5")];
597
- tensor<fp16, [1, 1920, 32, 32]> input_147_cast_fp16 = concat(axis = var_2929, interleave = input_147_interleave_0, values = (hidden_states_95_cast_fp16, cast_8))[name = string("input_147_cast_fp16")];
598
  tensor<int32, [5]> reshape_60_shape_0 = const()[name = string("reshape_60_shape_0"), val = tensor<int32, [5]>([1, 32, 60, 32, 32])];
599
  tensor<fp16, [1, 32, 60, 32, 32]> reshape_60_cast_fp16 = reshape(shape = reshape_60_shape_0, x = input_147_cast_fp16)[name = string("reshape_60_cast_fp16")];
600
  tensor<int32, [3]> reduce_mean_45_axes_0 = const()[name = string("reduce_mean_45_axes_0"), val = tensor<int32, [3]>([2, 3, 4])];
@@ -631,7 +631,7 @@ program(1.3)
631
  int32 temb_11_groups_0 = const()[name = string("temb_11_groups_0"), val = int32(1)];
632
  tensor<fp16, [640, 1280, 1, 1]> up_blocks_1_resnets_0_time_emb_proj_weight_to_fp16 = const()[name = string("up_blocks_1_resnets_0_time_emb_proj_weight_to_fp16"), val = tensor<fp16, [640, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(219495488)))];
633
  tensor<fp16, [640]> up_blocks_1_resnets_0_time_emb_proj_bias_to_fp16 = const()[name = string("up_blocks_1_resnets_0_time_emb_proj_bias_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(221133952)))];
634
- tensor<fp16, [1, 640, 1, 1]> temb_11_cast_fp16 = conv(bias = up_blocks_1_resnets_0_time_emb_proj_bias_to_fp16, dilations = temb_11_dilations_0, groups = temb_11_groups_0, pad = temb_11_pad_0, pad_type = temb_11_pad_type_0, strides = temb_11_strides_0, weight = up_blocks_1_resnets_0_time_emb_proj_weight_to_fp16, x = cast_3)[name = string("temb_11_cast_fp16")];
635
  tensor<fp16, [1, 640, 32, 32]> input_155_cast_fp16 = add(x = hidden_states_97_cast_fp16, y = temb_11_cast_fp16)[name = string("input_155_cast_fp16")];
636
  tensor<int32, [5]> reshape_64_shape_0 = const()[name = string("reshape_64_shape_0"), val = tensor<int32, [5]>([1, 32, 20, 32, 32])];
637
  tensor<fp16, [1, 32, 20, 32, 32]> reshape_64_cast_fp16 = reshape(shape = reshape_64_shape_0, x = input_155_cast_fp16)[name = string("reshape_64_cast_fp16")];
@@ -1123,8 +1123,8 @@ program(1.3)
1123
  tensor<fp16, [1, 640, 32, 32]> hidden_states_113_cast_fp16 = conv(bias = up_blocks_1_attentions_0_proj_out_bias_to_fp16, dilations = hidden_states_113_dilations_0, groups = hidden_states_113_groups_0, pad = hidden_states_113_pad_0, pad_type = hidden_states_113_pad_type_0, strides = hidden_states_113_strides_0, weight = up_blocks_1_attentions_0_proj_out_weight_to_fp16, x = input_171_cast_fp16)[name = string("hidden_states_113_cast_fp16")];
1124
  tensor<fp16, [1, 640, 32, 32]> hidden_states_115_cast_fp16 = add(x = hidden_states_113_cast_fp16, y = hidden_states_101_cast_fp16)[name = string("hidden_states_115_cast_fp16")];
1125
  bool input_173_interleave_0 = const()[name = string("input_173_interleave_0"), val = bool(false)];
1126
- tensor<fp16, [1, 320, 32, 32]> cast_7 = cast(dtype = cast_7_dtype_0, x = input_37_cast_fp16)[name = string("cast_6")];
1127
- tensor<fp16, [1, 960, 32, 32]> input_173_cast_fp16 = concat(axis = var_2929, interleave = input_173_interleave_0, values = (hidden_states_115_cast_fp16, cast_7))[name = string("input_173_cast_fp16")];
1128
  tensor<int32, [5]> reshape_72_shape_0 = const()[name = string("reshape_72_shape_0"), val = tensor<int32, [5]>([1, 32, 30, 32, 32])];
1129
  tensor<fp16, [1, 32, 30, 32, 32]> reshape_72_cast_fp16 = reshape(shape = reshape_72_shape_0, x = input_173_cast_fp16)[name = string("reshape_72_cast_fp16")];
1130
  tensor<int32, [3]> reduce_mean_54_axes_0 = const()[name = string("reduce_mean_54_axes_0"), val = tensor<int32, [3]>([2, 3, 4])];
@@ -1163,7 +1163,7 @@ program(1.3)
1163
  int32 temb_13_groups_0 = const()[name = string("temb_13_groups_0"), val = int32(1)];
1164
  tensor<fp16, [640, 1280, 1, 1]> up_blocks_1_resnets_1_time_emb_proj_weight_to_fp16 = const()[name = string("up_blocks_1_resnets_1_time_emb_proj_weight_to_fp16"), val = tensor<fp16, [640, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(260418368)))];
1165
  tensor<fp16, [640]> up_blocks_1_resnets_1_time_emb_proj_bias_to_fp16 = const()[name = string("up_blocks_1_resnets_1_time_emb_proj_bias_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(262056832)))];
1166
- tensor<fp16, [1, 640, 1, 1]> temb_13_cast_fp16 = conv(bias = up_blocks_1_resnets_1_time_emb_proj_bias_to_fp16, dilations = temb_13_dilations_0, groups = temb_13_groups_0, pad = temb_13_pad_0, pad_type = temb_13_pad_type_0, strides = temb_13_strides_0, weight = up_blocks_1_resnets_1_time_emb_proj_weight_to_fp16, x = cast_3)[name = string("temb_13_cast_fp16")];
1167
  tensor<fp16, [1, 640, 32, 32]> input_181_cast_fp16 = add(x = hidden_states_117_cast_fp16, y = temb_13_cast_fp16)[name = string("input_181_cast_fp16")];
1168
  tensor<int32, [5]> reshape_76_shape_0 = const()[name = string("reshape_76_shape_0"), val = tensor<int32, [5]>([1, 32, 20, 32, 32])];
1169
  tensor<fp16, [1, 32, 20, 32, 32]> reshape_76_cast_fp16 = reshape(shape = reshape_76_shape_0, x = input_181_cast_fp16)[name = string("reshape_76_cast_fp16")];
@@ -1667,8 +1667,8 @@ program(1.3)
1667
  tensor<fp16, [1, 640, 64, 64]> hidden_states_135_cast_fp16 = conv(bias = up_blocks_1_upsamplers_0_conv_bias_to_fp16, dilations = hidden_states_135_dilations_0, groups = hidden_states_135_groups_0, pad = hidden_states_135_pad_0, pad_type = hidden_states_135_pad_type_0, strides = hidden_states_135_strides_0, weight = up_blocks_1_upsamplers_0_conv_weight_to_fp16, x = input_201_cast_fp16)[name = string("hidden_states_135_cast_fp16")];
1668
  int32 var_4045 = const()[name = string("op_4045"), val = int32(1)];
1669
  bool input_203_interleave_0 = const()[name = string("input_203_interleave_0"), val = bool(false)];
1670
- tensor<fp16, [1, 320, 64, 64]> cast_5 = cast(dtype = cast_5_dtype_0, x = input_35_cast_fp16)[name = string("cast_7")];
1671
- tensor<fp16, [1, 960, 64, 64]> input_203_cast_fp16 = concat(axis = var_4045, interleave = input_203_interleave_0, values = (hidden_states_135_cast_fp16, cast_5))[name = string("input_203_cast_fp16")];
1672
  tensor<int32, [5]> reshape_84_shape_0 = const()[name = string("reshape_84_shape_0"), val = tensor<int32, [5]>([1, 32, 30, 64, 64])];
1673
  tensor<fp16, [1, 32, 30, 64, 64]> reshape_84_cast_fp16 = reshape(shape = reshape_84_shape_0, x = input_203_cast_fp16)[name = string("reshape_84_cast_fp16")];
1674
  tensor<int32, [3]> reduce_mean_63_axes_0 = const()[name = string("reduce_mean_63_axes_0"), val = tensor<int32, [3]>([2, 3, 4])];
@@ -1705,7 +1705,7 @@ program(1.3)
1705
  int32 temb_15_groups_0 = const()[name = string("temb_15_groups_0"), val = int32(1)];
1706
  tensor<fp16, [320, 1280, 1, 1]> up_blocks_2_resnets_0_time_emb_proj_weight_to_fp16 = const()[name = string("up_blocks_2_resnets_0_time_emb_proj_weight_to_fp16"), val = tensor<fp16, [320, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(301952448)))];
1707
  tensor<fp16, [320]> up_blocks_2_resnets_0_time_emb_proj_bias_to_fp16 = const()[name = string("up_blocks_2_resnets_0_time_emb_proj_bias_to_fp16"), val = tensor<fp16, [320]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(302771712)))];
1708
- tensor<fp16, [1, 320, 1, 1]> temb_15_cast_fp16 = conv(bias = up_blocks_2_resnets_0_time_emb_proj_bias_to_fp16, dilations = temb_15_dilations_0, groups = temb_15_groups_0, pad = temb_15_pad_0, pad_type = temb_15_pad_type_0, strides = temb_15_strides_0, weight = up_blocks_2_resnets_0_time_emb_proj_weight_to_fp16, x = cast_3)[name = string("temb_15_cast_fp16")];
1709
  tensor<fp16, [1, 320, 64, 64]> input_211_cast_fp16 = add(x = hidden_states_137_cast_fp16, y = temb_15_cast_fp16)[name = string("input_211_cast_fp16")];
1710
  tensor<int32, [5]> reshape_88_shape_0 = const()[name = string("reshape_88_shape_0"), val = tensor<int32, [5]>([1, 32, 10, 64, 64])];
1711
  tensor<fp16, [1, 32, 10, 64, 64]> reshape_88_cast_fp16 = reshape(shape = reshape_88_shape_0, x = input_211_cast_fp16)[name = string("reshape_88_cast_fp16")];
@@ -2197,8 +2197,8 @@ program(1.3)
2197
  tensor<fp16, [1, 320, 64, 64]> hidden_states_153_cast_fp16 = conv(bias = up_blocks_2_attentions_0_proj_out_bias_to_fp16, dilations = hidden_states_153_dilations_0, groups = hidden_states_153_groups_0, pad = hidden_states_153_pad_0, pad_type = hidden_states_153_pad_type_0, strides = hidden_states_153_strides_0, weight = up_blocks_2_attentions_0_proj_out_weight_to_fp16, x = input_227_cast_fp16)[name = string("hidden_states_153_cast_fp16")];
2198
  tensor<fp16, [1, 320, 64, 64]> hidden_states_155_cast_fp16 = add(x = hidden_states_153_cast_fp16, y = hidden_states_141_cast_fp16)[name = string("hidden_states_155_cast_fp16")];
2199
  bool input_229_interleave_0 = const()[name = string("input_229_interleave_0"), val = bool(false)];
2200
- tensor<fp16, [1, 320, 64, 64]> cast_0 = cast(dtype = cast_0_dtype_0, x = input_7_cast_fp16)[name = string("cast_9")];
2201
- tensor<fp16, [1, 640, 64, 64]> input_229_cast_fp16 = concat(axis = var_4045, interleave = input_229_interleave_0, values = (hidden_states_155_cast_fp16, cast_0))[name = string("input_229_cast_fp16")];
2202
  tensor<int32, [5]> reshape_96_shape_0 = const()[name = string("reshape_96_shape_0"), val = tensor<int32, [5]>([1, 32, 20, 64, 64])];
2203
  tensor<fp16, [1, 32, 20, 64, 64]> reshape_96_cast_fp16 = reshape(shape = reshape_96_shape_0, x = input_229_cast_fp16)[name = string("reshape_96_cast_fp16")];
2204
  tensor<int32, [3]> reduce_mean_72_axes_0 = const()[name = string("reduce_mean_72_axes_0"), val = tensor<int32, [3]>([2, 3, 4])];
@@ -2235,7 +2235,7 @@ program(1.3)
2235
  int32 temb_groups_0 = const()[name = string("temb_groups_0"), val = int32(1)];
2236
  tensor<fp16, [320, 1280, 1, 1]> up_blocks_2_resnets_1_time_emb_proj_weight_to_fp16 = const()[name = string("up_blocks_2_resnets_1_time_emb_proj_weight_to_fp16"), val = tensor<fp16, [320, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(314016960)))];
2237
  tensor<fp16, [320]> up_blocks_2_resnets_1_time_emb_proj_bias_to_fp16 = const()[name = string("up_blocks_2_resnets_1_time_emb_proj_bias_to_fp16"), val = tensor<fp16, [320]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(314836224)))];
2238
- tensor<fp16, [1, 320, 1, 1]> temb_cast_fp16 = conv(bias = up_blocks_2_resnets_1_time_emb_proj_bias_to_fp16, dilations = temb_dilations_0, groups = temb_groups_0, pad = temb_pad_0, pad_type = temb_pad_type_0, strides = temb_strides_0, weight = up_blocks_2_resnets_1_time_emb_proj_weight_to_fp16, x = cast_3)[name = string("temb_cast_fp16")];
2239
  tensor<fp16, [1, 320, 64, 64]> input_237_cast_fp16 = add(x = hidden_states_157_cast_fp16, y = temb_cast_fp16)[name = string("input_237_cast_fp16")];
2240
  tensor<int32, [5]> reshape_100_shape_0 = const()[name = string("reshape_100_shape_0"), val = tensor<int32, [5]>([1, 32, 10, 64, 64])];
2241
  tensor<fp16, [1, 32, 10, 64, 64]> reshape_100_cast_fp16 = reshape(shape = reshape_100_shape_0, x = input_237_cast_fp16)[name = string("reshape_100_cast_fp16")];
@@ -2753,7 +2753,7 @@ program(1.3)
2753
  tensor<int32, [2]> var_5145_dilations_0 = const()[name = string("op_5145_dilations_0"), val = tensor<int32, [2]>([1, 1])];
2754
  int32 var_5145_groups_0 = const()[name = string("op_5145_groups_0"), val = int32(1)];
2755
  tensor<fp16, [4, 320, 3, 3]> conv_out_weight_to_fp16 = const()[name = string("conv_out_weight_to_fp16"), val = tensor<fp16, [4, 320, 3, 3]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(322188224)))];
2756
- tensor<fp16, [4]> conv_out_bias_to_fp16 = const()[name = string("conv_out_bias_to_fp16"), val = tensor<fp16, [4]>([0x1.a24p-10, -0x1.a8p-11, 0x1.21p-11, -0x1.39cp-10])];
2757
  tensor<fp16, [1, 4, 64, 64]> var_5145_cast_fp16 = conv(bias = conv_out_bias_to_fp16, dilations = var_5145_dilations_0, groups = var_5145_groups_0, pad = var_5145_pad_0, pad_type = var_5145_pad_type_0, strides = var_5145_strides_0, weight = conv_out_weight_to_fp16, x = input_cast_fp16)[name = string("op_5145_cast_fp16")];
2758
  string var_5145_cast_fp16_to_fp32_dtype_0 = const()[name = string("op_5145_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
2759
  tensor<fp32, [1, 4, 64, 64]> noise_pred = cast(dtype = var_5145_cast_fp16_to_fp32_dtype_0, x = var_5145_cast_fp16)[name = string("cast_0")];
 
2
  [buildInfo = dict<string, string>({{"coremlc-component-MIL", "3600.16.1"}, {"coremlc-version", "3600.22.1"}})]
3
  {
4
  func main<ios18>(tensor<fp16, [1, 768, 1, 77]> encoder_hidden_states, tensor<fp32, [1, 1280, 16, 16]> hidden_states_61_cast_fp16, tensor<fp32, [1, 1280, 1, 1]> input_15_cast_fp16, tensor<fp32, [1, 320, 64, 64]> input_35_cast_fp16, tensor<fp32, [1, 320, 32, 32]> input_37_cast_fp16, tensor<fp32, [1, 640, 32, 32]> input_63_cast_fp16, tensor<fp32, [1, 640, 16, 16]> input_65_cast_fp16, tensor<fp32, [1, 320, 64, 64]> input_7_cast_fp16, tensor<fp32, [1, 1280, 1, 256]> inputs_23_cast_fp16, tensor<fp32, [1, 10240, 1, 256]> var_2337_cast_fp16) {
5
+ string cast_8_dtype_0 = const()[name = string("cast_8_dtype_0"), val = string("fp16")];
6
  tensor<fp16, [320]> add_1_mean_0_to_fp16 = const()[name = string("add_1_mean_0_to_fp16"), val = tensor<fp16, [320]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
7
  tensor<fp16, [320]> add_1_variance_0_to_fp16 = const()[name = string("add_1_variance_0_to_fp16"), val = tensor<fp16, [320]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(768)))];
 
 
8
  string cast_7_dtype_0 = const()[name = string("cast_7_dtype_0"), val = string("fp16")];
9
+ string cast_2_dtype_0 = const()[name = string("cast_2_dtype_0"), val = string("fp16")];
10
+ string cast_5_dtype_0 = const()[name = string("cast_5_dtype_0"), val = string("fp16")];
11
  tensor<fp16, [640]> add_9_mean_0_to_fp16 = const()[name = string("add_9_mean_0_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1472)))];
12
  tensor<fp16, [640]> add_9_variance_0_to_fp16 = const()[name = string("add_9_variance_0_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2816)))];
13
+ string cast_6_dtype_0 = const()[name = string("cast_6_dtype_0"), val = string("fp16")];
14
+ string cast_0_dtype_0 = const()[name = string("cast_0_dtype_0"), val = string("fp16")];
15
  tensor<fp16, [1280]> add_15_mean_0_to_fp16 = const()[name = string("add_15_mean_0_to_fp16"), val = tensor<fp16, [1280]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4160)))];
16
  tensor<fp16, [1280]> add_15_variance_0_to_fp16 = const()[name = string("add_15_variance_0_to_fp16"), val = tensor<fp16, [1280]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6784)))];
17
  int32 var_1812 = const()[name = string("op_1812"), val = int32(1)];
18
+ string cast_3_dtype_0 = const()[name = string("cast_3_dtype_0"), val = string("fp16")];
19
+ string cast_1_dtype_0 = const()[name = string("cast_1_dtype_0"), val = string("fp16")];
20
  string cast_4_dtype_0 = const()[name = string("cast_4_dtype_0"), val = string("fp16")];
 
 
21
  tensor<int32, [2]> var_2338_split_sizes_0 = const()[name = string("op_2338_split_sizes_0"), val = tensor<int32, [2]>([5120, 5120])];
22
  int32 var_2338_axis_0 = const()[name = string("op_2338_axis_0"), val = int32(1)];
23
+ tensor<fp16, [1, 10240, 1, 256]> cast_4 = cast(dtype = cast_4_dtype_0, x = var_2337_cast_fp16)[name = string("cast_1")];
24
+ tensor<fp16, [1, 5120, 1, 256]> var_2338_cast_fp16_0, tensor<fp16, [1, 5120, 1, 256]> var_2338_cast_fp16_1 = split(axis = var_2338_axis_0, split_sizes = var_2338_split_sizes_0, x = cast_4)[name = string("op_2338_cast_fp16")];
25
  string var_2340_mode_0 = const()[name = string("op_2340_mode_0"), val = string("EXACT")];
26
  tensor<fp16, [1, 5120, 1, 256]> var_2340_cast_fp16 = gelu(mode = var_2340_mode_0, x = var_2338_cast_fp16_1)[name = string("op_2340_cast_fp16")];
27
  tensor<fp16, [1, 5120, 1, 256]> input_113_cast_fp16 = mul(x = var_2338_cast_fp16_0, y = var_2340_cast_fp16)[name = string("input_113_cast_fp16")];
 
33
  tensor<fp16, [1280, 5120, 1, 1]> up_blocks_0_attentions_0_transformer_blocks_0_ff_net_2_weight_to_fp16 = const()[name = string("up_blocks_0_attentions_0_transformer_blocks_0_ff_net_2_weight_to_fp16"), val = tensor<fp16, [1280, 5120, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(9408)))];
34
  tensor<fp16, [1280]> up_blocks_0_attentions_0_transformer_blocks_0_ff_net_2_bias_to_fp16 = const()[name = string("up_blocks_0_attentions_0_transformer_blocks_0_ff_net_2_bias_to_fp16"), val = tensor<fp16, [1280]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13116672)))];
35
  tensor<fp16, [1, 1280, 1, 256]> var_2348_cast_fp16 = conv(bias = up_blocks_0_attentions_0_transformer_blocks_0_ff_net_2_bias_to_fp16, dilations = var_2348_dilations_0, groups = var_2348_groups_0, pad = var_2348_pad_0, pad_type = var_2348_pad_type_0, strides = var_2348_strides_0, weight = up_blocks_0_attentions_0_transformer_blocks_0_ff_net_2_weight_to_fp16, x = input_113_cast_fp16)[name = string("op_2348_cast_fp16")];
36
+ tensor<fp16, [1, 1280, 1, 256]> cast_1 = cast(dtype = cast_1_dtype_0, x = inputs_23_cast_fp16)[name = string("cast_2")];
37
+ tensor<fp16, [1, 1280, 1, 256]> hidden_states_71_cast_fp16 = add(x = var_2348_cast_fp16, y = cast_1)[name = string("hidden_states_71_cast_fp16")];
38
  tensor<int32, [4]> var_2350 = const()[name = string("op_2350"), val = tensor<int32, [4]>([1, 1280, 16, 16])];
39
  tensor<fp16, [1, 1280, 16, 16]> input_115_cast_fp16 = reshape(shape = var_2350, x = hidden_states_71_cast_fp16)[name = string("input_115_cast_fp16")];
40
  string hidden_states_73_pad_type_0 = const()[name = string("hidden_states_73_pad_type_0"), val = string("valid")];
 
45
  tensor<fp16, [1280, 1280, 1, 1]> up_blocks_0_attentions_0_proj_out_weight_to_fp16 = const()[name = string("up_blocks_0_attentions_0_proj_out_weight_to_fp16"), val = tensor<fp16, [1280, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13119296)))];
46
  tensor<fp16, [1280]> up_blocks_0_attentions_0_proj_out_bias_to_fp16 = const()[name = string("up_blocks_0_attentions_0_proj_out_bias_to_fp16"), val = tensor<fp16, [1280]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(16396160)))];
47
  tensor<fp16, [1, 1280, 16, 16]> hidden_states_73_cast_fp16 = conv(bias = up_blocks_0_attentions_0_proj_out_bias_to_fp16, dilations = hidden_states_73_dilations_0, groups = hidden_states_73_groups_0, pad = hidden_states_73_pad_0, pad_type = hidden_states_73_pad_type_0, strides = hidden_states_73_strides_0, weight = up_blocks_0_attentions_0_proj_out_weight_to_fp16, x = input_115_cast_fp16)[name = string("hidden_states_73_cast_fp16")];
48
+ tensor<fp16, [1, 1280, 16, 16]> cast_3 = cast(dtype = cast_3_dtype_0, x = hidden_states_61_cast_fp16)[name = string("cast_3")];
49
+ tensor<fp16, [1, 1280, 16, 16]> hidden_states_75_cast_fp16 = add(x = hidden_states_73_cast_fp16, y = cast_3)[name = string("hidden_states_75_cast_fp16")];
50
  bool input_117_interleave_0 = const()[name = string("input_117_interleave_0"), val = bool(false)];
51
+ tensor<fp16, [1, 640, 16, 16]> cast_0 = cast(dtype = cast_0_dtype_0, x = input_65_cast_fp16)[name = string("cast_4")];
52
+ tensor<fp16, [1, 1920, 16, 16]> input_117_cast_fp16 = concat(axis = var_1812, interleave = input_117_interleave_0, values = (hidden_states_75_cast_fp16, cast_0))[name = string("input_117_cast_fp16")];
53
  tensor<int32, [5]> reshape_48_shape_0 = const()[name = string("reshape_48_shape_0"), val = tensor<int32, [5]>([1, 32, 60, 16, 16])];
54
  tensor<fp16, [1, 32, 60, 16, 16]> reshape_48_cast_fp16 = reshape(shape = reshape_48_shape_0, x = input_117_cast_fp16)[name = string("reshape_48_cast_fp16")];
55
  tensor<int32, [3]> reduce_mean_36_axes_0 = const()[name = string("reduce_mean_36_axes_0"), val = tensor<int32, [3]>([2, 3, 4])];
 
88
  int32 temb_9_groups_0 = const()[name = string("temb_9_groups_0"), val = int32(1)];
89
  tensor<fp16, [1280, 1280, 1, 1]> up_blocks_0_resnets_1_time_emb_proj_weight_to_fp16 = const()[name = string("up_blocks_0_resnets_1_time_emb_proj_weight_to_fp16"), val = tensor<fp16, [1280, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(60653888)))];
90
  tensor<fp16, [1280]> up_blocks_0_resnets_1_time_emb_proj_bias_to_fp16 = const()[name = string("up_blocks_0_resnets_1_time_emb_proj_bias_to_fp16"), val = tensor<fp16, [1280]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(63930752)))];
91
+ tensor<fp16, [1, 1280, 1, 1]> cast_7 = cast(dtype = cast_7_dtype_0, x = input_15_cast_fp16)[name = string("cast_8")];
92
+ tensor<fp16, [1, 1280, 1, 1]> temb_9_cast_fp16 = conv(bias = up_blocks_0_resnets_1_time_emb_proj_bias_to_fp16, dilations = temb_9_dilations_0, groups = temb_9_groups_0, pad = temb_9_pad_0, pad_type = temb_9_pad_type_0, strides = temb_9_strides_0, weight = up_blocks_0_resnets_1_time_emb_proj_weight_to_fp16, x = cast_7)[name = string("temb_9_cast_fp16")];
93
  tensor<fp16, [1, 1280, 16, 16]> input_125_cast_fp16 = add(x = hidden_states_77_cast_fp16, y = temb_9_cast_fp16)[name = string("input_125_cast_fp16")];
94
  tensor<int32, [5]> reshape_52_shape_0 = const()[name = string("reshape_52_shape_0"), val = tensor<int32, [5]>([1, 32, 40, 16, 16])];
95
  tensor<fp16, [1, 32, 40, 16, 16]> reshape_52_cast_fp16 = reshape(shape = reshape_52_shape_0, x = input_125_cast_fp16)[name = string("reshape_52_cast_fp16")];
 
593
  tensor<fp16, [1, 1280, 32, 32]> hidden_states_95_cast_fp16 = conv(bias = up_blocks_0_upsamplers_0_conv_bias_to_fp16, dilations = hidden_states_95_dilations_0, groups = hidden_states_95_groups_0, pad = hidden_states_95_pad_0, pad_type = hidden_states_95_pad_type_0, strides = hidden_states_95_strides_0, weight = up_blocks_0_upsamplers_0_conv_weight_to_fp16, x = input_145_cast_fp16)[name = string("hidden_states_95_cast_fp16")];
594
  int32 var_2929 = const()[name = string("op_2929"), val = int32(1)];
595
  bool input_147_interleave_0 = const()[name = string("input_147_interleave_0"), val = bool(false)];
596
+ tensor<fp16, [1, 640, 32, 32]> cast_6 = cast(dtype = cast_6_dtype_0, x = input_63_cast_fp16)[name = string("cast_5")];
597
+ tensor<fp16, [1, 1920, 32, 32]> input_147_cast_fp16 = concat(axis = var_2929, interleave = input_147_interleave_0, values = (hidden_states_95_cast_fp16, cast_6))[name = string("input_147_cast_fp16")];
598
  tensor<int32, [5]> reshape_60_shape_0 = const()[name = string("reshape_60_shape_0"), val = tensor<int32, [5]>([1, 32, 60, 32, 32])];
599
  tensor<fp16, [1, 32, 60, 32, 32]> reshape_60_cast_fp16 = reshape(shape = reshape_60_shape_0, x = input_147_cast_fp16)[name = string("reshape_60_cast_fp16")];
600
  tensor<int32, [3]> reduce_mean_45_axes_0 = const()[name = string("reduce_mean_45_axes_0"), val = tensor<int32, [3]>([2, 3, 4])];
 
631
  int32 temb_11_groups_0 = const()[name = string("temb_11_groups_0"), val = int32(1)];
632
  tensor<fp16, [640, 1280, 1, 1]> up_blocks_1_resnets_0_time_emb_proj_weight_to_fp16 = const()[name = string("up_blocks_1_resnets_0_time_emb_proj_weight_to_fp16"), val = tensor<fp16, [640, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(219495488)))];
633
  tensor<fp16, [640]> up_blocks_1_resnets_0_time_emb_proj_bias_to_fp16 = const()[name = string("up_blocks_1_resnets_0_time_emb_proj_bias_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(221133952)))];
634
+ tensor<fp16, [1, 640, 1, 1]> temb_11_cast_fp16 = conv(bias = up_blocks_1_resnets_0_time_emb_proj_bias_to_fp16, dilations = temb_11_dilations_0, groups = temb_11_groups_0, pad = temb_11_pad_0, pad_type = temb_11_pad_type_0, strides = temb_11_strides_0, weight = up_blocks_1_resnets_0_time_emb_proj_weight_to_fp16, x = cast_7)[name = string("temb_11_cast_fp16")];
635
  tensor<fp16, [1, 640, 32, 32]> input_155_cast_fp16 = add(x = hidden_states_97_cast_fp16, y = temb_11_cast_fp16)[name = string("input_155_cast_fp16")];
636
  tensor<int32, [5]> reshape_64_shape_0 = const()[name = string("reshape_64_shape_0"), val = tensor<int32, [5]>([1, 32, 20, 32, 32])];
637
  tensor<fp16, [1, 32, 20, 32, 32]> reshape_64_cast_fp16 = reshape(shape = reshape_64_shape_0, x = input_155_cast_fp16)[name = string("reshape_64_cast_fp16")];
 
1123
  tensor<fp16, [1, 640, 32, 32]> hidden_states_113_cast_fp16 = conv(bias = up_blocks_1_attentions_0_proj_out_bias_to_fp16, dilations = hidden_states_113_dilations_0, groups = hidden_states_113_groups_0, pad = hidden_states_113_pad_0, pad_type = hidden_states_113_pad_type_0, strides = hidden_states_113_strides_0, weight = up_blocks_1_attentions_0_proj_out_weight_to_fp16, x = input_171_cast_fp16)[name = string("hidden_states_113_cast_fp16")];
1124
  tensor<fp16, [1, 640, 32, 32]> hidden_states_115_cast_fp16 = add(x = hidden_states_113_cast_fp16, y = hidden_states_101_cast_fp16)[name = string("hidden_states_115_cast_fp16")];
1125
  bool input_173_interleave_0 = const()[name = string("input_173_interleave_0"), val = bool(false)];
1126
+ tensor<fp16, [1, 320, 32, 32]> cast_5 = cast(dtype = cast_5_dtype_0, x = input_37_cast_fp16)[name = string("cast_6")];
1127
+ tensor<fp16, [1, 960, 32, 32]> input_173_cast_fp16 = concat(axis = var_2929, interleave = input_173_interleave_0, values = (hidden_states_115_cast_fp16, cast_5))[name = string("input_173_cast_fp16")];
1128
  tensor<int32, [5]> reshape_72_shape_0 = const()[name = string("reshape_72_shape_0"), val = tensor<int32, [5]>([1, 32, 30, 32, 32])];
1129
  tensor<fp16, [1, 32, 30, 32, 32]> reshape_72_cast_fp16 = reshape(shape = reshape_72_shape_0, x = input_173_cast_fp16)[name = string("reshape_72_cast_fp16")];
1130
  tensor<int32, [3]> reduce_mean_54_axes_0 = const()[name = string("reduce_mean_54_axes_0"), val = tensor<int32, [3]>([2, 3, 4])];
 
1163
  int32 temb_13_groups_0 = const()[name = string("temb_13_groups_0"), val = int32(1)];
1164
  tensor<fp16, [640, 1280, 1, 1]> up_blocks_1_resnets_1_time_emb_proj_weight_to_fp16 = const()[name = string("up_blocks_1_resnets_1_time_emb_proj_weight_to_fp16"), val = tensor<fp16, [640, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(260418368)))];
1165
  tensor<fp16, [640]> up_blocks_1_resnets_1_time_emb_proj_bias_to_fp16 = const()[name = string("up_blocks_1_resnets_1_time_emb_proj_bias_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(262056832)))];
1166
+ tensor<fp16, [1, 640, 1, 1]> temb_13_cast_fp16 = conv(bias = up_blocks_1_resnets_1_time_emb_proj_bias_to_fp16, dilations = temb_13_dilations_0, groups = temb_13_groups_0, pad = temb_13_pad_0, pad_type = temb_13_pad_type_0, strides = temb_13_strides_0, weight = up_blocks_1_resnets_1_time_emb_proj_weight_to_fp16, x = cast_7)[name = string("temb_13_cast_fp16")];
1167
  tensor<fp16, [1, 640, 32, 32]> input_181_cast_fp16 = add(x = hidden_states_117_cast_fp16, y = temb_13_cast_fp16)[name = string("input_181_cast_fp16")];
1168
  tensor<int32, [5]> reshape_76_shape_0 = const()[name = string("reshape_76_shape_0"), val = tensor<int32, [5]>([1, 32, 20, 32, 32])];
1169
  tensor<fp16, [1, 32, 20, 32, 32]> reshape_76_cast_fp16 = reshape(shape = reshape_76_shape_0, x = input_181_cast_fp16)[name = string("reshape_76_cast_fp16")];
 
1667
  tensor<fp16, [1, 640, 64, 64]> hidden_states_135_cast_fp16 = conv(bias = up_blocks_1_upsamplers_0_conv_bias_to_fp16, dilations = hidden_states_135_dilations_0, groups = hidden_states_135_groups_0, pad = hidden_states_135_pad_0, pad_type = hidden_states_135_pad_type_0, strides = hidden_states_135_strides_0, weight = up_blocks_1_upsamplers_0_conv_weight_to_fp16, x = input_201_cast_fp16)[name = string("hidden_states_135_cast_fp16")];
1668
  int32 var_4045 = const()[name = string("op_4045"), val = int32(1)];
1669
  bool input_203_interleave_0 = const()[name = string("input_203_interleave_0"), val = bool(false)];
1670
+ tensor<fp16, [1, 320, 64, 64]> cast_2 = cast(dtype = cast_2_dtype_0, x = input_35_cast_fp16)[name = string("cast_7")];
1671
+ tensor<fp16, [1, 960, 64, 64]> input_203_cast_fp16 = concat(axis = var_4045, interleave = input_203_interleave_0, values = (hidden_states_135_cast_fp16, cast_2))[name = string("input_203_cast_fp16")];
1672
  tensor<int32, [5]> reshape_84_shape_0 = const()[name = string("reshape_84_shape_0"), val = tensor<int32, [5]>([1, 32, 30, 64, 64])];
1673
  tensor<fp16, [1, 32, 30, 64, 64]> reshape_84_cast_fp16 = reshape(shape = reshape_84_shape_0, x = input_203_cast_fp16)[name = string("reshape_84_cast_fp16")];
1674
  tensor<int32, [3]> reduce_mean_63_axes_0 = const()[name = string("reduce_mean_63_axes_0"), val = tensor<int32, [3]>([2, 3, 4])];
 
1705
  int32 temb_15_groups_0 = const()[name = string("temb_15_groups_0"), val = int32(1)];
1706
  tensor<fp16, [320, 1280, 1, 1]> up_blocks_2_resnets_0_time_emb_proj_weight_to_fp16 = const()[name = string("up_blocks_2_resnets_0_time_emb_proj_weight_to_fp16"), val = tensor<fp16, [320, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(301952448)))];
1707
  tensor<fp16, [320]> up_blocks_2_resnets_0_time_emb_proj_bias_to_fp16 = const()[name = string("up_blocks_2_resnets_0_time_emb_proj_bias_to_fp16"), val = tensor<fp16, [320]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(302771712)))];
1708
+ tensor<fp16, [1, 320, 1, 1]> temb_15_cast_fp16 = conv(bias = up_blocks_2_resnets_0_time_emb_proj_bias_to_fp16, dilations = temb_15_dilations_0, groups = temb_15_groups_0, pad = temb_15_pad_0, pad_type = temb_15_pad_type_0, strides = temb_15_strides_0, weight = up_blocks_2_resnets_0_time_emb_proj_weight_to_fp16, x = cast_7)[name = string("temb_15_cast_fp16")];
1709
  tensor<fp16, [1, 320, 64, 64]> input_211_cast_fp16 = add(x = hidden_states_137_cast_fp16, y = temb_15_cast_fp16)[name = string("input_211_cast_fp16")];
1710
  tensor<int32, [5]> reshape_88_shape_0 = const()[name = string("reshape_88_shape_0"), val = tensor<int32, [5]>([1, 32, 10, 64, 64])];
1711
  tensor<fp16, [1, 32, 10, 64, 64]> reshape_88_cast_fp16 = reshape(shape = reshape_88_shape_0, x = input_211_cast_fp16)[name = string("reshape_88_cast_fp16")];
 
2197
  tensor<fp16, [1, 320, 64, 64]> hidden_states_153_cast_fp16 = conv(bias = up_blocks_2_attentions_0_proj_out_bias_to_fp16, dilations = hidden_states_153_dilations_0, groups = hidden_states_153_groups_0, pad = hidden_states_153_pad_0, pad_type = hidden_states_153_pad_type_0, strides = hidden_states_153_strides_0, weight = up_blocks_2_attentions_0_proj_out_weight_to_fp16, x = input_227_cast_fp16)[name = string("hidden_states_153_cast_fp16")];
2198
  tensor<fp16, [1, 320, 64, 64]> hidden_states_155_cast_fp16 = add(x = hidden_states_153_cast_fp16, y = hidden_states_141_cast_fp16)[name = string("hidden_states_155_cast_fp16")];
2199
  bool input_229_interleave_0 = const()[name = string("input_229_interleave_0"), val = bool(false)];
2200
+ tensor<fp16, [1, 320, 64, 64]> cast_8 = cast(dtype = cast_8_dtype_0, x = input_7_cast_fp16)[name = string("cast_9")];
2201
+ tensor<fp16, [1, 640, 64, 64]> input_229_cast_fp16 = concat(axis = var_4045, interleave = input_229_interleave_0, values = (hidden_states_155_cast_fp16, cast_8))[name = string("input_229_cast_fp16")];
2202
  tensor<int32, [5]> reshape_96_shape_0 = const()[name = string("reshape_96_shape_0"), val = tensor<int32, [5]>([1, 32, 20, 64, 64])];
2203
  tensor<fp16, [1, 32, 20, 64, 64]> reshape_96_cast_fp16 = reshape(shape = reshape_96_shape_0, x = input_229_cast_fp16)[name = string("reshape_96_cast_fp16")];
2204
  tensor<int32, [3]> reduce_mean_72_axes_0 = const()[name = string("reduce_mean_72_axes_0"), val = tensor<int32, [3]>([2, 3, 4])];
 
2235
  int32 temb_groups_0 = const()[name = string("temb_groups_0"), val = int32(1)];
2236
  tensor<fp16, [320, 1280, 1, 1]> up_blocks_2_resnets_1_time_emb_proj_weight_to_fp16 = const()[name = string("up_blocks_2_resnets_1_time_emb_proj_weight_to_fp16"), val = tensor<fp16, [320, 1280, 1, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(314016960)))];
2237
  tensor<fp16, [320]> up_blocks_2_resnets_1_time_emb_proj_bias_to_fp16 = const()[name = string("up_blocks_2_resnets_1_time_emb_proj_bias_to_fp16"), val = tensor<fp16, [320]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(314836224)))];
2238
+ tensor<fp16, [1, 320, 1, 1]> temb_cast_fp16 = conv(bias = up_blocks_2_resnets_1_time_emb_proj_bias_to_fp16, dilations = temb_dilations_0, groups = temb_groups_0, pad = temb_pad_0, pad_type = temb_pad_type_0, strides = temb_strides_0, weight = up_blocks_2_resnets_1_time_emb_proj_weight_to_fp16, x = cast_7)[name = string("temb_cast_fp16")];
2239
  tensor<fp16, [1, 320, 64, 64]> input_237_cast_fp16 = add(x = hidden_states_157_cast_fp16, y = temb_cast_fp16)[name = string("input_237_cast_fp16")];
2240
  tensor<int32, [5]> reshape_100_shape_0 = const()[name = string("reshape_100_shape_0"), val = tensor<int32, [5]>([1, 32, 10, 64, 64])];
2241
  tensor<fp16, [1, 32, 10, 64, 64]> reshape_100_cast_fp16 = reshape(shape = reshape_100_shape_0, x = input_237_cast_fp16)[name = string("reshape_100_cast_fp16")];
 
2753
  tensor<int32, [2]> var_5145_dilations_0 = const()[name = string("op_5145_dilations_0"), val = tensor<int32, [2]>([1, 1])];
2754
  int32 var_5145_groups_0 = const()[name = string("op_5145_groups_0"), val = int32(1)];
2755
  tensor<fp16, [4, 320, 3, 3]> conv_out_weight_to_fp16 = const()[name = string("conv_out_weight_to_fp16"), val = tensor<fp16, [4, 320, 3, 3]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(322188224)))];
2756
+ tensor<fp16, [4]> conv_out_bias_to_fp16 = const()[name = string("conv_out_bias_to_fp16"), val = tensor<fp16, [4]>([0x1.d64p-10, -0x1.31cp-11, 0x1.548p-12, -0x1.674p-10])];
2757
  tensor<fp16, [1, 4, 64, 64]> var_5145_cast_fp16 = conv(bias = conv_out_bias_to_fp16, dilations = var_5145_dilations_0, groups = var_5145_groups_0, pad = var_5145_pad_0, pad_type = var_5145_pad_type_0, strides = var_5145_strides_0, weight = conv_out_weight_to_fp16, x = input_cast_fp16)[name = string("op_5145_cast_fp16")];
2758
  string var_5145_cast_fp16_to_fp32_dtype_0 = const()[name = string("op_5145_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
2759
  tensor<fp32, [1, 4, 64, 64]> noise_pred = cast(dtype = var_5145_cast_fp16_to_fp32_dtype_0, x = var_5145_cast_fp16)[name = string("cast_0")];
UnetChunk2.mlmodelc/weights/weight.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:06720754c9b2d5f1165a7989f07be3cb8a879f68363b27f85820afb47ce9353d
3
  size 322211328
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:04910ba6d4ca88b34dd1202c848aebfc33e339010d203573ebc4138a4e9f78ba
3
  size 322211328