From b83b472748a51ca4db1a88513cd7626f81aaa490 Mon Sep 17 00:00:00 2001
From: lirui <771725652@qq.com>
Date: Sun, 5 Jan 2025 14:40:25 +0800
Subject: [PATCH] Fix some args

---
 xfuser/config/args.py   | 2 +-
 xfuser/config/config.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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):