Skip to content

Instantly share code, notes, and snippets.

@josephrocca

josephrocca/lyra-v2-soundstream-tflite-to-onnx-conversion-error - IndexError list index out of range.ipynb

Last active October 9, 2022 04:22
Show Gist options
  • Save josephrocca/0581baa153235adf9a3e6560203edc80 to your computer and use it in GitHub Desktop.
Save josephrocca/0581baa153235adf9a3e6560203edc80 to your computer and use it in GitHub Desktop.
Download ZIP
lyra-v2-soundstream-tflite-to-onnx-conversion-error-indexerror-list-index-out-of-range.ipynb
Display the source blob
Display the rendered blob
Raw
lyra-v2-soundstream-tflite-to-onnx-conversion-error - IndexError list index out of range.ipynb
{
"nbformat": 4,
"nbformat_minor": 0,
"metadata": {
"colab": {
"provenance": [],
"collapsed_sections": [],
"authorship_tag": "ABX9TyNXqhgz4fRfe/ICu9LDaFtV",
"include_colab_link": true
},
"kernelspec": {
"name": "python3",
"display_name": "Python 3"
},
"language_info": {
"name": "python"
},
"accelerator": "GPU",
"gpuClass": "standard"
},
"cells": [
{
"cell_type": "markdown",
"metadata": {
"id": "view-in-github",
"colab_type": "text"
},
"source": [
"<a href=\"https://colab.research.google.com/gist/josephrocca/0581baa153235adf9a3e6560203edc80/lyra-v2-soundstream-tflite-to-onnx-conversion-error-didn-t-find-op-for-builtin-opcode-conv_2d-version-6.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
]
},
{
"cell_type": "code",
"source": [
"!pip install tensorflow==2.9.2"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "4YHd4pEU14zL",
"outputId": "5bdea30b-bd95-4f00-8de5-d1a81b18bf3a"
},
"execution_count": 1,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
"Collecting tensorflow==2.9.2\n",
" Downloading tensorflow-2.9.2-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (511.8 MB)\n",
"\u001b[K |████████████████████████████████| 511.8 MB 9.5 kB/s \n",
"\u001b[?25hRequirement already satisfied: protobuf<3.20,>=3.9.2 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (3.17.3)\n",
"Requirement already satisfied: h5py>=2.9.0 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (3.1.0)\n",
"Requirement already satisfied: termcolor>=1.1.0 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (2.0.1)\n",
"Requirement already satisfied: typing-extensions>=3.6.6 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (4.1.1)\n",
"Requirement already satisfied: packaging in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (21.3)\n",
"Requirement already satisfied: google-pasta>=0.1.1 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (0.2.0)\n",
"Collecting tensorboard<2.10,>=2.9\n",
" Downloading tensorboard-2.9.1-py3-none-any.whl (5.8 MB)\n",
"\u001b[K |████████████████████████████████| 5.8 MB 54.9 MB/s \n",
"\u001b[?25hRequirement already satisfied: setuptools in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (57.4.0)\n",
"Requirement already satisfied: keras-preprocessing>=1.1.1 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (1.1.2)\n",
"Requirement already satisfied: libclang>=13.0.0 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (14.0.6)\n",
"Collecting tensorflow-estimator<2.10.0,>=2.9.0rc0\n",
" Downloading tensorflow_estimator-2.9.0-py2.py3-none-any.whl (438 kB)\n",
"\u001b[K |████████████████████████████████| 438 kB 70.0 MB/s \n",
"\u001b[?25hRequirement already satisfied: absl-py>=1.0.0 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (1.2.0)\n",
"Requirement already satisfied: opt-einsum>=2.3.2 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (3.3.0)\n",
"Requirement already satisfied: numpy>=1.20 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (1.21.6)\n",
"Collecting gast<=0.4.0,>=0.2.1\n",
" Downloading gast-0.4.0-py3-none-any.whl (9.8 kB)\n",
"Collecting flatbuffers<2,>=1.12\n",
" Downloading flatbuffers-1.12-py2.py3-none-any.whl (15 kB)\n",
"Requirement already satisfied: astunparse>=1.6.0 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (1.6.3)\n",
"Collecting keras<2.10.0,>=2.9.0rc0\n",
" Downloading keras-2.9.0-py2.py3-none-any.whl (1.6 MB)\n",
"\u001b[K |████████████████████████████████| 1.6 MB 57.1 MB/s \n",
"\u001b[?25hRequirement already satisfied: six>=1.12.0 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (1.15.0)\n",
"Requirement already satisfied: grpcio<2.0,>=1.24.3 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (1.49.1)\n",
"Requirement already satisfied: tensorflow-io-gcs-filesystem>=0.23.1 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (0.27.0)\n",
"Requirement already satisfied: wrapt>=1.11.0 in /usr/local/lib/python3.7/dist-packages (from tensorflow==2.9.2) (1.14.1)\n",
"Requirement already satisfied: wheel<1.0,>=0.23.0 in /usr/local/lib/python3.7/dist-packages (from astunparse>=1.6.0->tensorflow==2.9.2) (0.37.1)\n",
"Requirement already satisfied: cached-property in /usr/local/lib/python3.7/dist-packages (from h5py>=2.9.0->tensorflow==2.9.2) (1.5.2)\n",
"Requirement already satisfied: werkzeug>=1.0.1 in /usr/local/lib/python3.7/dist-packages (from tensorboard<2.10,>=2.9->tensorflow==2.9.2) (1.0.1)\n",
"Requirement already satisfied: markdown>=2.6.8 in /usr/local/lib/python3.7/dist-packages (from tensorboard<2.10,>=2.9->tensorflow==2.9.2) (3.4.1)\n",
"Requirement already satisfied: requests<3,>=2.21.0 in /usr/local/lib/python3.7/dist-packages (from tensorboard<2.10,>=2.9->tensorflow==2.9.2) (2.23.0)\n",
"Requirement already satisfied: tensorboard-plugin-wit>=1.6.0 in /usr/local/lib/python3.7/dist-packages (from tensorboard<2.10,>=2.9->tensorflow==2.9.2) (1.8.1)\n",
"Requirement already satisfied: google-auth<3,>=1.6.3 in /usr/local/lib/python3.7/dist-packages (from tensorboard<2.10,>=2.9->tensorflow==2.9.2) (1.35.0)\n",
"Requirement already satisfied: google-auth-oauthlib<0.5,>=0.4.1 in /usr/local/lib/python3.7/dist-packages (from tensorboard<2.10,>=2.9->tensorflow==2.9.2) (0.4.6)\n",
"Requirement already satisfied: tensorboard-data-server<0.7.0,>=0.6.0 in /usr/local/lib/python3.7/dist-packages (from tensorboard<2.10,>=2.9->tensorflow==2.9.2) (0.6.1)\n",
"Requirement already satisfied: pyasn1-modules>=0.2.1 in /usr/local/lib/python3.7/dist-packages (from google-auth<3,>=1.6.3->tensorboard<2.10,>=2.9->tensorflow==2.9.2) (0.2.8)\n",
"Requirement already satisfied: rsa<5,>=3.1.4 in /usr/local/lib/python3.7/dist-packages (from google-auth<3,>=1.6.3->tensorboard<2.10,>=2.9->tensorflow==2.9.2) (4.9)\n",
"Requirement already satisfied: cachetools<5.0,>=2.0.0 in /usr/local/lib/python3.7/dist-packages (from google-auth<3,>=1.6.3->tensorboard<2.10,>=2.9->tensorflow==2.9.2) (4.2.4)\n",
"Requirement already satisfied: requests-oauthlib>=0.7.0 in /usr/local/lib/python3.7/dist-packages (from google-auth-oauthlib<0.5,>=0.4.1->tensorboard<2.10,>=2.9->tensorflow==2.9.2) (1.3.1)\n",
"Requirement already satisfied: importlib-metadata>=4.4 in /usr/local/lib/python3.7/dist-packages (from markdown>=2.6.8->tensorboard<2.10,>=2.9->tensorflow==2.9.2) (5.0.0)\n",
"Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.7/dist-packages (from importlib-metadata>=4.4->markdown>=2.6.8->tensorboard<2.10,>=2.9->tensorflow==2.9.2) (3.8.1)\n",
"Requirement already satisfied: pyasn1<0.5.0,>=0.4.6 in /usr/local/lib/python3.7/dist-packages (from pyasn1-modules>=0.2.1->google-auth<3,>=1.6.3->tensorboard<2.10,>=2.9->tensorflow==2.9.2) (0.4.8)\n",
"Requirement already satisfied: idna<3,>=2.5 in /usr/local/lib/python3.7/dist-packages (from requests<3,>=2.21.0->tensorboard<2.10,>=2.9->tensorflow==2.9.2) (2.10)\n",
"Requirement already satisfied: chardet<4,>=3.0.2 in /usr/local/lib/python3.7/dist-packages (from requests<3,>=2.21.0->tensorboard<2.10,>=2.9->tensorflow==2.9.2) (3.0.4)\n",
"Requirement already satisfied: urllib3!=1.25.0,!=1.25.1,<1.26,>=1.21.1 in /usr/local/lib/python3.7/dist-packages (from requests<3,>=2.21.0->tensorboard<2.10,>=2.9->tensorflow==2.9.2) (1.24.3)\n",
"Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.7/dist-packages (from requests<3,>=2.21.0->tensorboard<2.10,>=2.9->tensorflow==2.9.2) (2022.9.24)\n",
"Requirement already satisfied: oauthlib>=3.0.0 in /usr/local/lib/python3.7/dist-packages (from requests-oauthlib>=0.7.0->google-auth-oauthlib<0.5,>=0.4.1->tensorboard<2.10,>=2.9->tensorflow==2.9.2) (3.2.1)\n",
"Requirement already satisfied: pyparsing!=3.0.5,>=2.0.2 in /usr/local/lib/python3.7/dist-packages (from packaging->tensorflow==2.9.2) (3.0.9)\n",
"Installing collected packages: tensorflow-estimator, tensorboard, keras, gast, flatbuffers, tensorflow\n",
" Attempting uninstall: tensorflow-estimator\n",
" Found existing installation: tensorflow-estimator 2.8.0\n",
" Uninstalling tensorflow-estimator-2.8.0:\n",
" Successfully uninstalled tensorflow-estimator-2.8.0\n",
" Attempting uninstall: tensorboard\n",
" Found existing installation: tensorboard 2.8.0\n",
" Uninstalling tensorboard-2.8.0:\n",
" Successfully uninstalled tensorboard-2.8.0\n",
" Attempting uninstall: keras\n",
" Found existing installation: keras 2.8.0\n",
" Uninstalling keras-2.8.0:\n",
" Successfully uninstalled keras-2.8.0\n",
" Attempting uninstall: gast\n",
" Found existing installation: gast 0.5.3\n",
" Uninstalling gast-0.5.3:\n",
" Successfully uninstalled gast-0.5.3\n",
" Attempting uninstall: flatbuffers\n",
" Found existing installation: flatbuffers 22.9.24\n",
" Uninstalling flatbuffers-22.9.24:\n",
" Successfully uninstalled flatbuffers-22.9.24\n",
" Attempting uninstall: tensorflow\n",
" Found existing installation: tensorflow 2.8.2+zzzcolab20220929150707\n",
" Uninstalling tensorflow-2.8.2+zzzcolab20220929150707:\n",
" Successfully uninstalled tensorflow-2.8.2+zzzcolab20220929150707\n",
"Successfully installed flatbuffers-1.12 gast-0.4.0 keras-2.9.0 tensorboard-2.9.1 tensorflow-2.9.2 tensorflow-estimator-2.9.0\n"
]
}
]
},
{
"cell_type": "code",
"source": [
"import tensorflow as tf\n",
"tf.__version__"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 35
},
"id": "ndU-oI3x3P-_",
"outputId": "1a71a161-d9db-418b-95b6-d3b36b2d064b"
},
"execution_count": 2,
"outputs": [
{
"output_type": "execute_result",
"data": {
"text/plain": [
"'2.9.2'"
],
"application/vnd.google.colaboratory.intrinsic+json": {
"type": "string"
}
},
"metadata": {},
"execution_count": 2
}
]
},
{
"cell_type": "code",
"source": [
"!pip install git+https://github.com/onnx/tensorflow-onnx"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "_Our00mFGN_J",
"outputId": "a15e219a-6e69-43b8-d42d-5a36881bfa75"
},
"execution_count": 3,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n",
"Collecting git+https://github.com/onnx/tensorflow-onnx\n",
" Cloning https://github.com/onnx/tensorflow-onnx to /tmp/pip-req-build-voqc02f8\n",
" Running command git clone -q https://github.com/onnx/tensorflow-onnx /tmp/pip-req-build-voqc02f8\n",
"Requirement already satisfied: numpy>=1.14.1 in /usr/local/lib/python3.7/dist-packages (from tf2onnx==1.12.0) (1.21.6)\n",
"Collecting onnx>=1.4.1\n",
" Downloading onnx-1.12.0-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (13.1 MB)\n",
"\u001b[K |████████████████████████████████| 13.1 MB 5.0 MB/s \n",
"\u001b[?25hRequirement already satisfied: requests in /usr/local/lib/python3.7/dist-packages (from tf2onnx==1.12.0) (2.23.0)\n",
"Requirement already satisfied: six in /usr/local/lib/python3.7/dist-packages (from tf2onnx==1.12.0) (1.15.0)\n",
"Requirement already satisfied: flatbuffers>=1.12 in /usr/local/lib/python3.7/dist-packages (from tf2onnx==1.12.0) (1.12)\n",
"Requirement already satisfied: protobuf<=3.20.1,>=3.12.2 in /usr/local/lib/python3.7/dist-packages (from onnx>=1.4.1->tf2onnx==1.12.0) (3.17.3)\n",
"Requirement already satisfied: typing-extensions>=3.6.2.1 in /usr/local/lib/python3.7/dist-packages (from onnx>=1.4.1->tf2onnx==1.12.0) (4.1.1)\n",
"Requirement already satisfied: chardet<4,>=3.0.2 in /usr/local/lib/python3.7/dist-packages (from requests->tf2onnx==1.12.0) (3.0.4)\n",
"Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.7/dist-packages (from requests->tf2onnx==1.12.0) (2022.9.24)\n",
"Requirement already satisfied: urllib3!=1.25.0,!=1.25.1,<1.26,>=1.21.1 in /usr/local/lib/python3.7/dist-packages (from requests->tf2onnx==1.12.0) (1.24.3)\n",
"Requirement already satisfied: idna<3,>=2.5 in /usr/local/lib/python3.7/dist-packages (from requests->tf2onnx==1.12.0) (2.10)\n",
"Building wheels for collected packages: tf2onnx\n",
" Building wheel for tf2onnx (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
" Created wheel for tf2onnx: filename=tf2onnx-1.12.0-py3-none-any.whl size=442139 sha256=30f384f069617bceaf5f861c2b670720ca45dd0362e12d596212510a291aa000\n",
" Stored in directory: /tmp/pip-ephem-wheel-cache-l58ntg49/wheels/73/99/8e/3034c9b47124da4ac24a6ea72b8da1be5af878e37c2e953bb3\n",
"Successfully built tf2onnx\n",
"Installing collected packages: onnx, tf2onnx\n",
"Successfully installed onnx-1.12.0 tf2onnx-1.12.0\n"
]
}
]
},
{
"cell_type": "code",
"source": [
"!wget https://huggingface.co/rocca/lyra-v2-soundstream/resolve/main/tflite/soundstream_encoder.tflite"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "go1tTAjRF3KO",
"outputId": "04d560ce-7cd0-4b31-85a7-6dd44a1abbbc"
},
"execution_count": 4,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"--2022-10-09 03:58:14-- https://huggingface.co/rocca/lyra-v2-soundstream/resolve/main/tflite/soundstream_encoder.tflite\n",
"Resolving huggingface.co (huggingface.co)... 54.210.225.113, 44.195.102.200, 54.173.5.192, ...\n",
"Connecting to huggingface.co (huggingface.co)|54.210.225.113|:443... connected.\n",
"HTTP request sent, awaiting response... 302 Found\n",
"Location: https://cdn-lfs.huggingface.co/repos/1f/fe/1ffe1c47407cdb0e4d0ffd7a0a45e55975087b2366351b5ffa9063ec95d6d1f4/6110a3f31ea11a46af8686042bce725b2f0da91b5e88e3cbbed004288698ee53?response-content-disposition=attachment%3B%20filename%3D%22soundstream_encoder.tflite%22&Expires=1665546530&Policy=eyJTdGF0ZW1lbnQiOlt7IlJlc291cmNlIjoiaHR0cHM6Ly9jZG4tbGZzLmh1Z2dpbmdmYWNlLmNvL3JlcG9zLzFmL2ZlLzFmZmUxYzQ3NDA3Y2RiMGU0ZDBmZmQ3YTBhNDVlNTU5NzUwODdiMjM2NjM1MWI1ZmZhOTA2M2VjOTVkNmQxZjQvNjExMGEzZjMxZWExMWE0NmFmODY4NjA0MmJjZTcyNWIyZjBkYTkxYjVlODhlM2NiYmVkMDA0Mjg4Njk4ZWU1Mz9yZXNwb25zZS1jb250ZW50LWRpc3Bvc2l0aW9uPWF0dGFjaG1lbnQlM0IlMjBmaWxlbmFtZSUzRCUyMnNvdW5kc3RyZWFtX2VuY29kZXIudGZsaXRlJTIyIiwiQ29uZGl0aW9uIjp7IkRhdGVMZXNzVGhhbiI6eyJBV1M6RXBvY2hUaW1lIjoxNjY1NTQ2NTMwfX19XX0_&Signature=PmZiRrqRgVyvX~~iGy41N4~m-0Rj4~A0ggZqRUo6V15bNtcZsZkBRZBXB65SPLNJY2pPjPvHqatovEyPLqXjZkqSWWHeG5h5~4-6M~ab1MVGcarDTx4VVZJropOmXAQm7kGNhLgW5uEMsCVZ6QbN6PEbjL9GKoWPMHsiOFM~w7yif4~D4omsbqIDG-dOK0zyXrH~S1zECBugWEw6lRmLo70jcSli7Cwe4kzahE2nY5yCz9l0ZIO~UPBLuk8EDFBQ-xfn25DmAi1moF~k81RZGFEl1YQui7Fwm6El3pL0JJvPNByRYgDfaY0pgQSN6hvChW5jUwbtMI7eSzTpbX6S1Q__&Key-Pair-Id=KVTP0A1DKRTAX [following]\n",
"--2022-10-09 03:58:14-- https://cdn-lfs.huggingface.co/repos/1f/fe/1ffe1c47407cdb0e4d0ffd7a0a45e55975087b2366351b5ffa9063ec95d6d1f4/6110a3f31ea11a46af8686042bce725b2f0da91b5e88e3cbbed004288698ee53?response-content-disposition=attachment%3B%20filename%3D%22soundstream_encoder.tflite%22&Expires=1665546530&Policy=eyJTdGF0ZW1lbnQiOlt7IlJlc291cmNlIjoiaHR0cHM6Ly9jZG4tbGZzLmh1Z2dpbmdmYWNlLmNvL3JlcG9zLzFmL2ZlLzFmZmUxYzQ3NDA3Y2RiMGU0ZDBmZmQ3YTBhNDVlNTU5NzUwODdiMjM2NjM1MWI1ZmZhOTA2M2VjOTVkNmQxZjQvNjExMGEzZjMxZWExMWE0NmFmODY4NjA0MmJjZTcyNWIyZjBkYTkxYjVlODhlM2NiYmVkMDA0Mjg4Njk4ZWU1Mz9yZXNwb25zZS1jb250ZW50LWRpc3Bvc2l0aW9uPWF0dGFjaG1lbnQlM0IlMjBmaWxlbmFtZSUzRCUyMnNvdW5kc3RyZWFtX2VuY29kZXIudGZsaXRlJTIyIiwiQ29uZGl0aW9uIjp7IkRhdGVMZXNzVGhhbiI6eyJBV1M6RXBvY2hUaW1lIjoxNjY1NTQ2NTMwfX19XX0_&Signature=PmZiRrqRgVyvX~~iGy41N4~m-0Rj4~A0ggZqRUo6V15bNtcZsZkBRZBXB65SPLNJY2pPjPvHqatovEyPLqXjZkqSWWHeG5h5~4-6M~ab1MVGcarDTx4VVZJropOmXAQm7kGNhLgW5uEMsCVZ6QbN6PEbjL9GKoWPMHsiOFM~w7yif4~D4omsbqIDG-dOK0zyXrH~S1zECBugWEw6lRmLo70jcSli7Cwe4kzahE2nY5yCz9l0ZIO~UPBLuk8EDFBQ-xfn25DmAi1moF~k81RZGFEl1YQui7Fwm6El3pL0JJvPNByRYgDfaY0pgQSN6hvChW5jUwbtMI7eSzTpbX6S1Q__&Key-Pair-Id=KVTP0A1DKRTAX\n",
"Resolving cdn-lfs.huggingface.co (cdn-lfs.huggingface.co)... 54.230.18.21, 54.230.18.124, 54.230.18.98, ...\n",
"Connecting to cdn-lfs.huggingface.co (cdn-lfs.huggingface.co)|54.230.18.21|:443... connected.\n",
"HTTP request sent, awaiting response... 200 OK\n",
"Length: 2748716 (2.6M) [binary/octet-stream]\n",
"Saving to: ‘soundstream_encoder.tflite’\n",
"\n",
"soundstream_encoder 100%[===================>] 2.62M --.-KB/s in 0.09s \n",
"\n",
"2022-10-09 03:58:15 (30.3 MB/s) - ‘soundstream_encoder.tflite’ saved [2748716/2748716]\n",
"\n"
]
}
]
},
{
"cell_type": "code",
"execution_count": 11,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "jjQIusn9FvAR",
"outputId": "f90c0b6b-42b7-4328-bd17-263e7e9d52fa"
},
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"/usr/lib/python3.7/runpy.py:125: RuntimeWarning: 'tf2onnx.convert' found in sys.modules after import of package 'tf2onnx', but prior to execution of 'tf2onnx.convert'; this may result in unpredictable behaviour\n",
" warn(RuntimeWarning(msg))\n",
"2022-10-09 04:08:11,357 - INFO - tf2onnx: inputs: None\n",
"2022-10-09 04:08:11,357 - INFO - tf2onnx: outputs: None\n",
"2022-10-09 04:08:12,242 - INFO - tf2onnx.tfonnx: Using tensorflow=2.9.2, onnx=1.12.0, tf2onnx=1.12.0/c60c5e\n",
"2022-10-09 04:08:12,243 - INFO - tf2onnx.tfonnx: Using opset <onnx, 16>\n",
"INFO: Created TensorFlow Lite XNNPACK delegate for CPU.\n",
"Traceback (most recent call last):\n",
" File \"/usr/lib/python3.7/runpy.py\", line 193, in _run_module_as_main\n",
" \"__main__\", mod_spec)\n",
" File \"/usr/lib/python3.7/runpy.py\", line 85, in _run_code\n",
" exec(code, run_globals)\n",
" File \"/usr/local/lib/python3.7/dist-packages/tf2onnx/convert.py\", line 706, in <module>\n",
" main()\n",
" File \"/usr/local/lib/python3.7/dist-packages/tf2onnx/convert.py\", line 291, in main\n",
" output_path=args.output)\n",
" File \"/usr/local/lib/python3.7/dist-packages/tf2onnx/convert.py\", line 165, in _convert_common\n",
" custom_op_handlers=custom_op_handlers, **kwargs)\n",
" File \"/usr/local/lib/python3.7/dist-packages/tf2onnx/tfonnx.py\", line 453, in process_tf_graph\n",
" main_g, subgraphs = graphs_from_tflite(tflite_path, input_names, output_names)\n",
" File \"/usr/local/lib/python3.7/dist-packages/tf2onnx/tflite_utils.py\", line 153, in graphs_from_tflite\n",
" parse_tflite_graph(tfl_graph, opcodes, model, prefix, tensor_shapes_from_interpreter)\n",
" File \"/usr/local/lib/python3.7/dist-packages/tf2onnx/tflite_utils.py\", line 485, in parse_tflite_graph\n",
" onnx_node = helper.make_node(optype, input_names, output_names, name=output_names[0], **attr)\n",
"IndexError: list index out of range\n"
]
}
],
"source": [
"!python -m tf2onnx.convert --opset 16 --tflite soundstream_encoder.tflite --output soundstream_encoder.onnx --verbose --debug # --inputs \"serving_default_input_audio:0\" --outputs \"StatefulPartitionedCall:0\""
]
}
]
}
Display the source blob
Display the rendered blob
Raw
lyra-v2-soundstream-tflite-to-onnx-conversion-error-indexerror-list-index-out-of-range.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment