Dymas/abogen
1
0
Fork 0
mirror of https://github.com/denizsafak/abogen.git synced 2026-07-18 13:40:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: Update Supertonic GPU configuration to include loader patching for ONNX providers

Browse Source
This commit is contained in:
JB
2025-12-21 08:58:51 -08:00
parent c2209eeb2a
commit 90786d2bed
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
abogen/tts_supertonic.py
+4 -1
View File
@@ -137,9 +137,12 @@ def _configure_supertonic_gpu() -> None:
providers.insert(0, "TensorrtExecutionProvider") providers.insert(0, "TensorrtExecutionProvider")
providers.append("CPUExecutionProvider") providers.append("CPUExecutionProvider")
# Patch supertonic's config before TTS import # Patch supertonic's config and loader before TTS import
# We must patch both because loader imports the value at module load time
import supertonic.config as supertonic_config import supertonic.config as supertonic_config
import supertonic.loader as supertonic_loader
supertonic_config.DEFAULT_ONNX_PROVIDERS = providers supertonic_config.DEFAULT_ONNX_PROVIDERS = providers
supertonic_loader.DEFAULT_ONNX_PROVIDERS = providers
logger.info("Supertonic ONNX providers configured: %s", providers) logger.info("Supertonic ONNX providers configured: %s", providers)
except Exception as exc: except Exception as exc:
logger.warning("Could not configure supertonic GPU providers: %s", exc) logger.warning("Could not configure supertonic GPU providers: %s", exc)