diff --git a/vall_e/config.py b/vall_e/config.py
index e455c05..f362397 100644
--- a/vall_e/config.py
+++ b/vall_e/config.py
@@ -13,10 +13,6 @@ class Config(ConfigBase):
     data_dirs: list[Path] = field(default_factory=lambda: [])
     test_data_dirs: list[Path] = field(default_factory=lambda: [])
 
-    batch_size: int = 24
-    eval_batch_size: int = 12
-    nj: int = 8
-
     @property
     def sample_rate(self):
         return 24_000
@@ -26,6 +22,7 @@ class Config(ConfigBase):
     token_dim: int = 256
     num_tokens: int = 1024
 
+    nj: int = 8
     batch_size: int = 128
     eval_batch_size: int = 512
     warmup_min_lr: float = 1e-6