From 9f87a74de09e38f9d8f3e7ebb5fd26fac44a3b0e Mon Sep 17 00:00:00 2001
From: Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>
Date: Wed, 22 Nov 2017 23:50:31 +0900
Subject: can remove aligner

---
 become_yukarin/model.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'become_yukarin/model.py')

diff --git a/become_yukarin/model.py b/become_yukarin/model.py
index 3b5102e..c475685 100644
--- a/become_yukarin/model.py
+++ b/become_yukarin/model.py
@@ -212,8 +212,18 @@ def create_predictor(config: ModelConfig):
 
 
 def create_aligner(config: ModelConfig):
+    assert config.enable_aligner
     aligner = Aligner(
         in_size=config.in_channels,
         out_time_length=config.aligner_out_time_length,
     )
     return aligner
+
+
+def create(config: ModelConfig):
+    predictor = create_predictor(config)
+    if config.enable_aligner:
+        aligner = create_aligner(config)
+    else:
+        aligner = None
+    return predictor, aligner
-- 
cgit v1.2.3-70-g09d2