msluszniak's picture
Core ML: drop the fp32 variant, fp16 is equivalent in task terms
2935033 verified
Raw
History Blame Contribute Delete
1.58 kB
{
"$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
"model": "selfie_segmentation",
"family": "selfie",
"capabilities": [
"semantic-segmentation"
],
"backend": "coreml",
"license": "apache-2.0",
"variants": [
{
"file": "selfie_segmentation_coreml_fp16.pte",
"precision": "fp16",
"quantized": false,
"default": true,
"methods": {
"forward": {
"inputs": [
{
"shape": [
1,
3,
256,
256
],
"dtype": "float32"
}
],
"outputs": [
{
"shape": [
1,
1,
256,
256
],
"dtype": "float32"
}
]
}
}
},
{
"file": "selfie_segmentation_landscape_coreml_fp16.pte",
"precision": "fp16",
"quantized": false,
"default": false,
"methods": {
"forward": {
"inputs": [
{
"shape": [
1,
3,
144,
256
],
"dtype": "float32"
}
],
"outputs": [
{
"shape": [
1,
1,
144,
256
],
"dtype": "float32"
}
]
}
}
}
]
}