seba's picture
Add VibeVoice CoreML models
d999afe
program(1.3)
[buildInfo = dict<string, string>({{"coremlc-component-MIL", "3405.2.1"}, {"coremlc-version", "3405.2.1"}, {"coremltools-component-milinternal", ""}, {"coremltools-version", "9.0"}})]
{
func main<ios18>(tensor<fp16, [1, 896, 1]> input_x) {
string var_14_pad_type_0 = const()[name = string("op_14_pad_type_0"), val = string("valid")];
tensor<int32, [1]> var_14_strides_0 = const()[name = string("op_14_strides_0"), val = tensor<int32, [1]>([1])];
tensor<int32, [2]> var_14_pad_0 = const()[name = string("op_14_pad_0"), val = tensor<int32, [2]>([0, 0])];
tensor<int32, [1]> var_14_dilations_0 = const()[name = string("op_14_dilations_0"), val = tensor<int32, [1]>([1])];
int32 var_14_groups_0 = const()[name = string("op_14_groups_0"), val = int32(1)];
tensor<fp16, [896, 896, 1]> var_8_to_fp16 = const()[name = string("op_8_to_fp16"), val = tensor<fp16, [896, 896, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
tensor<fp16, [896]> fc1_bias_to_fp16 = const()[name = string("fc1_bias_to_fp16"), val = tensor<fp16, [896]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1605760)))];
tensor<fp16, [1, 896, 1]> var_14_cast_fp16 = conv(bias = fc1_bias_to_fp16, dilations = var_14_dilations_0, groups = var_14_groups_0, pad = var_14_pad_0, pad_type = var_14_pad_type_0, strides = var_14_strides_0, weight = var_8_to_fp16, x = input_x)[name = string("op_14_cast_fp16")];
tensor<fp16, [1, 896, 1]> x_cast_fp16 = relu(x = var_14_cast_fp16)[name = string("x_cast_fp16")];
string var_26_pad_type_0 = const()[name = string("op_26_pad_type_0"), val = string("valid")];
tensor<int32, [1]> var_26_strides_0 = const()[name = string("op_26_strides_0"), val = tensor<int32, [1]>([1])];
tensor<int32, [2]> var_26_pad_0 = const()[name = string("op_26_pad_0"), val = tensor<int32, [2]>([0, 0])];
tensor<int32, [1]> var_26_dilations_0 = const()[name = string("op_26_dilations_0"), val = tensor<int32, [1]>([1])];
int32 var_26_groups_0 = const()[name = string("op_26_groups_0"), val = int32(1)];
tensor<fp16, [1, 896, 1]> var_20_to_fp16 = const()[name = string("op_20_to_fp16"), val = tensor<fp16, [1, 896, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1607616)))];
tensor<fp16, [1]> fc2_bias_to_fp16 = const()[name = string("fc2_bias_to_fp16"), val = tensor<fp16, [1]>([-0x1.36p-5])];
tensor<fp16, [1, 1, 1]> output = conv(bias = fc2_bias_to_fp16, dilations = var_26_dilations_0, groups = var_26_groups_0, pad = var_26_pad_0, pad_type = var_26_pad_type_0, strides = var_26_strides_0, weight = var_20_to_fp16, x = x_cast_fp16)[name = string("op_26_cast_fp16")];
} -> (output);
}