From e93e1488baacb82844b9e05061341cd9a5234fc0 Mon Sep 17 00:00:00 2001 From: Michael Hansen Date: Fri, 1 Apr 2022 11:49:20 -0400 Subject: [PATCH] Disable onnx optimizations on 32-bit ARM --- mimic3-tts/mimic3_tts/voice.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/mimic3-tts/mimic3_tts/voice.py b/mimic3-tts/mimic3_tts/voice.py index aa6ddce..df85146 100644 --- a/mimic3-tts/mimic3_tts/voice.py +++ b/mimic3-tts/mimic3_tts/voice.py @@ -15,6 +15,7 @@ # import csv import logging +import platform import time import typing from abc import ABCMeta, abstractmethod @@ -242,7 +243,15 @@ class Mimic3Voice(metaclass=ABCMeta): _LOGGER.debug("Loading model from %s", generator_path) # Load onnx model - session_options = session_options or onnxruntime.SessionOptions() + if session_options is None: + session_options = onnxruntime.SessionOptions() + + if platform.machine() == "armv7l": + # Enabling optimizations on 32-bit ARM crashes + session_options.graph_optimization_level = ( + onnxruntime.GraphOptimizationLevel.ORT_DISABLE_ALL + ) + onnx_model = onnxruntime.InferenceSession( str(generator_path), sess_options=session_options )