jyapayne/mimic3
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Disable onnx optimizations on 32-bit ARM

Browse source
This commit is contained in:
Michael Hansen 2022-04-01 11:49:20 -04:00
commit e93e1488ba
1 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
mimic3-tts/mimic3_tts/voice.py
View file

@ -15,6 +15,7 @@
# #
import csv import csv
import logging import logging
import platform
import time import time
import typing import typing
from abc import ABCMeta, abstractmethod from abc import ABCMeta, abstractmethod
@ -242,7 +243,15 @@ class Mimic3Voice(metaclass=ABCMeta):
_LOGGER.debug("Loading model from %s", generator_path) _LOGGER.debug("Loading model from %s", generator_path)
# Load onnx model # 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( onnx_model = onnxruntime.InferenceSession(
str(generator_path), sess_options=session_options str(generator_path), sess_options=session_options
) )