At the end of the process, only a .onnx file is downloaded, .tflite file is not available. #5765
Description
At the end of step 3, only the onnx file is available for download.
There are dependency issues during step 2 and other errors during step 3
What web browser you are using
Edge
The following is from "Step 2: Download Data"
ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
openwakeword 0.6.0 requires speexdsp-ns<1,>=0.1.2; platform_system == "Linux", which is not installed.
openwakeword 0.6.0 requires ai-edge-litert<3,>=2.0.2; platform_system == "Linux" or platform_system == "Darwin", but you have ai-edge-litert 1.4.0 which is incompatible.
ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
gcsfs 2025.3.0 requires fsspec==2025.3.0, but you have fsspec 2023.10.0 which is incompatible.
Successfully installed datasets-2.14.6 dill-0.3.7 fsspec-2023.10.0 multiprocess-0.70.15
WARNING: The following packages were previously imported in this runtime:
[dill]
You must restart the runtime in order to use newly installed versions.
The following is from "Step 3: Train the Model"
File "/content/openwakeword/openwakeword/train.py", line 901, in
convert_onnx_to_tflite(os.path.join(config["output_dir"], config["model_name"] + ".onnx"),
File "/content/openwakeword/openwakeword/train.py", line 578, in convert_onnx_to_tflite
from onnx_tf.backend import prepare
ModuleNotFoundError: No module named 'onnx_tf'
Traceback (most recent call last):
File "/usr/local/bin/onnx2tf", line 5, in
from onnx2tf import main
File "/usr/local/lib/python3.12/dist-packages/onnx2tf/init.py", line 1, in
from onnx2tf.onnx2tf import convert, main
File "/usr/local/lib/python3.12/dist-packages/onnx2tf/onnx2tf.py", line 39, in
import onnx_graphsurgeon as gs
File "/usr/local/lib/python3.12/dist-packages/onnx_graphsurgeon/init.py", line 1, in
from onnx_graphsurgeon.exporters.onnx_exporter import export_onnx
File "/usr/local/lib/python3.12/dist-packages/onnx_graphsurgeon/exporters/onnx_exporter.py", line 134, in
np.uint16, onnx.helper.float32_to_bfloat16
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: module 'onnx.helper' has no attribute 'float32_to_bfloat16'
mv: cannot stat 'my_custom_model/hey_frank_float32.tflite': No such file or directory
FileNotFoundError Traceback (most recent call last)
/tmp/ipython-input-3117315244.py in <cell line: 0>()
103
104 files.download(f"my_custom_model/{config['model_name']}.onnx")
--> 105 files.download(f"my_custom_model/{config['model_name']}.tflite")
/usr/local/lib/python3.12/dist-packages/google/colab/files.py in download(filename)
231 if not _os.path.exists(filename):
232 msg = 'Cannot find file: {}'.format(filename)
--> 233 raise FileNotFoundError(msg) # pylint: disable=undefined-variable
234
235 comm_manager = _IPython.get_ipython().kernel.comm_manager
FileNotFoundError: Cannot find file: my_custom_model/hey_frank.tflite