Update default configs

config/LibriTTS/ar-quarter.yml

@@ -0,0 +1,7 @@
+data_dirs: [data/LibriTTS/]
+spkr_name_getter: "lambda p: p.parts[-3]"
+
+model: ar-quarter
+batch_size: 48
+eval_batch_size: 48
+eval_every: 10_000

config/LibriTTS/ar.yml

@@ -1,7 +1,7 @@
-data_dirs: [data/LibriTTS/train-clean-100/]
+data_dirs: [data/LibriTTS/]
 spkr_name_getter: "lambda p: p.parts[-3]"
 
-model: ar-quarter
-batch_size: 8
-eval_batch_size: 8
+model: ar
+batch_size: 24
+eval_batch_size: 24
 eval_every: 10_000

config/LibriTTS/nar-quarter.yml

@@ -0,0 +1,6 @@
+data_dirs: [data/LibriTTS/]
+spkr_name_getter: "lambda p: p.parts[-3]"
+
+model: nar-quarter
+batch_size: 48
+eval_batch_size: 48

config/LibriTTS/nar.yml

@@ -1,6 +1,7 @@
-data_dirs: [data/LibriTTS/train-clean-100/]
+data_dirs: [data/LibriTTS/]
 spkr_name_getter: "lambda p: p.parts[-3]"
 
-model: nar-quarter
-batch_size: 8
-eval_batch_size: 8
+model: nar
+batch_size: 24
+eval_batch_size: 24
+eval_every: 1_000

vall_e/vall_e/__init__.py

@@ -27,14 +27,15 @@ def get_model(name):
             n_heads=8,
             n_layers=12,
         )
-    elif "-official" in name:
+    else:
+        if name not in ["ar", "nar"]:
+            raise NotImplementedError(name)
+
         model = Model(
             cfg.num_tokens,
             d_model=1024,
             n_heads=16,
             n_layers=12,
         )
-    else:
-        raise NotImplementedError(name)
 
     return model