diff --git a/config/LibriTTS/ar-quarter.yml b/config/LibriTTS/ar-quarter.yml
new file mode 100644
index 0000000..547905a
--- /dev/null
+++ b/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
diff --git a/config/LibriTTS/ar.yml b/config/LibriTTS/ar.yml
index 03cd1f4..85272d7 100644
--- a/config/LibriTTS/ar.yml
+++ b/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
diff --git a/config/LibriTTS/nar-quarter.yml b/config/LibriTTS/nar-quarter.yml
new file mode 100644
index 0000000..db005a8
--- /dev/null
+++ b/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
diff --git a/config/LibriTTS/nar.yml b/config/LibriTTS/nar.yml
index 997b31e..bbb529b 100644
--- a/config/LibriTTS/nar.yml
+++ b/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
diff --git a/vall_e/vall_e/__init__.py b/vall_e/vall_e/__init__.py
index f866af5..f593b35 100644
--- a/vall_e/vall_e/__init__.py
+++ b/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