| [ |
| { |
| "metadataOutputVersion" : "3.0", |
| "storagePrecision" : "Float16", |
| "outputSchema" : [ |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 2 × 640 × 64 × 64)", |
| "shortDescription" : "", |
| "shape" : "[2, 640, 64, 64]", |
| "name" : "input_113_cast_fp16", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 2 × 320 × 128 × 128)", |
| "shortDescription" : "", |
| "shape" : "[2, 320, 128, 128]", |
| "name" : "input_13_cast_fp16", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 2 × 1280 × 32 × 32)", |
| "shortDescription" : "", |
| "shape" : "[2, 1280, 32, 32]", |
| "name" : "hidden_states_281_cast_fp16", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 2 × 1280 × 32 × 32)", |
| "shortDescription" : "", |
| "shape" : "[2, 1280, 32, 32]", |
| "name" : "input_213_cast_fp16", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 2 × 2560 × 32 × 32)", |
| "shortDescription" : "", |
| "shape" : "[2, 2560, 32, 32]", |
| "name" : "input_423_cast_fp16", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 2 × 640 × 32 × 32)", |
| "shortDescription" : "", |
| "shape" : "[2, 640, 32, 32]", |
| "name" : "input_115_cast_fp16", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 2 × 640 × 64 × 64)", |
| "shortDescription" : "", |
| "shape" : "[2, 640, 64, 64]", |
| "name" : "input_79_cast_fp16", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 2 × 320 × 128 × 128)", |
| "shortDescription" : "", |
| "shape" : "[2, 320, 128, 128]", |
| "name" : "input_29_cast_fp16", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 2 × 320 × 128 × 128)", |
| "shortDescription" : "", |
| "shape" : "[2, 320, 128, 128]", |
| "name" : "input_43_cast_fp16", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 2 × 320 × 64 × 64)", |
| "shortDescription" : "", |
| "shape" : "[2, 320, 64, 64]", |
| "name" : "input_45_cast_fp16", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 2 × 1280 × 1 × 1)", |
| "shortDescription" : "", |
| "shape" : "[2, 1280, 1, 1]", |
| "name" : "input_21_cast_fp16", |
| "type" : "MultiArray" |
| } |
| ], |
| "modelParameters" : [ |
|
|
| ], |
| "specificationVersion" : 7, |
| "mlProgramOperationTypeHistogram" : { |
| "Transpose" : 68, |
| "Ios16.reduceMean" : 46, |
| "Ios16.sin" : 2, |
| "Ios16.softmax" : 1280, |
| "Split" : 34, |
| "Ios16.add" : 148, |
| "Concat" : 74, |
| "Ios16.realDiv" : 23, |
| "Ios16.square" : 23, |
| "ExpandDims" : 6, |
| "Ios16.sub" : 23, |
| "Ios16.sqrt" : 23, |
| "Ios16.einsum" : 2560, |
| "Ios16.conv" : 386, |
| "Ios16.reshape" : 58, |
| "Ios16.layerNorm" : 102, |
| "SliceByIndex" : 3844, |
| "Ios16.batchNorm" : 23, |
| "Ios16.silu" : 21, |
| "Ios16.gelu" : 34, |
| "Ios16.cast" : 11, |
| "Ios16.mul" : 1316, |
| "Ios16.cos" : 2 |
| }, |
| "computePrecision" : "Mixed (Float16, Float32, Int32)", |
| "isUpdatable" : "0", |
| "stateSchema" : [ |
|
|
| ], |
| "availability" : { |
| "macOS" : "13.0", |
| "tvOS" : "16.0", |
| "visionOS" : "1.0", |
| "watchOS" : "9.0", |
| "iOS" : "16.0", |
| "macCatalyst" : "16.0" |
| }, |
| "modelType" : { |
| "name" : "MLModelType_mlProgram" |
| }, |
| "userDefinedMetadata" : { |
|
|
| }, |
| "inputSchema" : [ |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 2 × 4 × 128 × 128)", |
| "shortDescription" : "", |
| "shape" : "[2, 4, 128, 128]", |
| "name" : "sample", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 2)", |
| "shortDescription" : "", |
| "shape" : "[2]", |
| "name" : "timestep", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 2 × 2048 × 1 × 77)", |
| "shortDescription" : "", |
| "shape" : "[2, 2048, 1, 77]", |
| "name" : "encoder_hidden_states", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 2 × 6)", |
| "shortDescription" : "", |
| "shape" : "[2, 6]", |
| "name" : "time_ids", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 2 × 1280)", |
| "shortDescription" : "", |
| "shape" : "[2, 1280]", |
| "name" : "text_embeds", |
| "type" : "MultiArray" |
| } |
| ], |
| "generatedClassName" : "Stable_Diffusion_version__Volumes_8TB_TB5_AI_Models_juggernaut_x_hyper_fp16_unet_chunk1", |
| "method" : "predict" |
| } |
| ] |
