From 3e5f87528d0a3c34a327d5789878138cf1326b38 Mon Sep 17 00:00:00 2001 From: Emerson Pedroso Date: Wed, 28 Aug 2024 03:53:01 -0300 Subject: [PATCH] loading custom model --- xphonebr/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/xphonebr/__init__.py b/xphonebr/__init__.py index 05ef727..d4d1939 100644 --- a/xphonebr/__init__.py +++ b/xphonebr/__init__.py @@ -18,6 +18,7 @@ class Phonemizer: def __init__(self, autoreg=False, normalizer=False, custom_model=None): self.model_path = self.download_model(autoreg) + # load custom model if custom_model: self.phones = ph.from_checkpoint(custom_model, device=device) else: