Fix some args

xdit-project · Jan 5, 2025 · b83b472 · b83b472
1 parent dd38452
commit b83b472
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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):