diff --git a/xfuser/config/args.py b/xfuser/config/args.py
index 06f0fb0..b2ae471 100644
--- a/xfuser/config/args.py
+++ b/xfuser/config/args.py
@@ -215,7 +215,7 @@ def add_cli_args(parser: FlexibleArgumentParser):
             "--vae_parallel_size",
             type=int,
             default=0,
-            help="VAE parallel size.",
+            help="Number of processes for VAE parallelization. 0: no seperate process for VAE, 1: run VAE in a separate process, >1: distribute VAE across multiple processes.",
         )
         parallel_group.add_argument(
             "--split_scheme",
diff --git a/xfuser/config/config.py b/xfuser/config/config.py
index 2e34c58..d224b67 100644
--- a/xfuser/config/config.py
+++ b/xfuser/config/config.py
@@ -193,7 +193,7 @@ class ParallelConfig:
     pp_config: PipeFusionParallelConfig
     tp_config: TensorParallelConfig
     world_size: int = 1 # FIXME: remove this
-    dit_world_size: int = 1 # FIXME: remove this
+    dit_world_size: int = 1
     vae_parallel_size: int = 1 # 0 means the vae is in the same process with diffusion
 
     def __post_init__(self):