loading custom model

traderpedroso · Aug 28, 2024 · 3e5f875 · 3e5f875
1 parent cc1fdeb
commit 3e5f875
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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: