xnnpack/config.json

{
  "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
  "model": "bk_sdm_tiny",
  "family": "bk_sdm",
  "capabilities": [
    "image-generation"
  ],
  "backend": "xnnpack",
  "license": "creativeml-openrail-m",
  "variants": [
    {
      "file": "bk_sdm_tiny_text_encoder_xnnpack_fp32.pte",
      "precision": "fp32",
      "quantized": false,
      "default": true,
      "methods": {
        "forward": {
          "inputs": [
            {
              "shape": [
                1,
                77
              ],
              "dtype": "int64"
            },
            {
              "shape": [
                1,
                77
              ],
              "dtype": "int64"
            }
          ],
          "outputs": [
            {
              "shape": [
                1,
                77,
                768
              ],
              "dtype": "float32"
            },
            {
              "shape": [
                1,
                768
              ],
              "dtype": "float32"
            }
          ]
        }
      }
    },
    {
      "file": "bk_sdm_tiny_unet_256_xnnpack_fp32.pte",
      "precision": "fp32",
      "quantized": false,
      "default": false,
      "methods": {
        "forward": {
          "inputs": [
            {
              "shape": [
                2,
                4,
                32,
                32
              ],
              "dtype": "float32"
            },
            {
              "shape": [
                1
              ],
              "dtype": "int64"
            },
            {
              "shape": [
                2,
                77,
                768
              ],
              "dtype": "float32"
            }
          ],
          "outputs": [
            {
              "shape": [
                2,
                4,
                32,
                32
              ],
              "dtype": "float32"
            }
          ]
        }
      }
    },
    {
      "file": "bk_sdm_tiny_unet_xnnpack_fp32.pte",
      "precision": "fp32",
      "quantized": false,
      "default": false,
      "methods": {
        "forward": {
          "inputs": [
            {
              "shape": [
                2,
                4,
                64,
                64
              ],
              "dtype": "float32"
            },
            {
              "shape": [
                1
              ],
              "dtype": "int64"
            },
            {
              "shape": [
                2,
                77,
                768
              ],
              "dtype": "float32"
            }
          ],
          "outputs": [
            {
              "shape": [
                2,
                4,
                64,
                64
              ],
              "dtype": "float32"
            }
          ]
        }
      }
    },
    {
      "file": "bk_sdm_tiny_vae_256_xnnpack_fp32.pte",
      "precision": "fp32",
      "quantized": false,
      "default": false,
      "methods": {
        "forward": {
          "inputs": [
            {
              "shape": [
                1,
                4,
                32,
                32
              ],
              "dtype": "float32"
            }
          ],
          "outputs": [
            {
              "shape": [
                256,
                256,
                3
              ],
              "dtype": "float32"
            }
          ]
        }
      }
    },
    {
      "file": "bk_sdm_tiny_vae_xnnpack_fp32.pte",
      "precision": "fp32",
      "quantized": false,
      "default": false,
      "methods": {
        "forward": {
          "inputs": [
            {
              "shape": [
                1,
                4,
                64,
                64
              ],
              "dtype": "float32"
            }
          ],
          "outputs": [
            {
              "shape": [
                512,
                512,
                3
              ],
              "dtype": "float32"
            }
          ]
        }
      }
    }
  ]
}