Skip to content

Commit

Permalink
use_parallel
Browse files Browse the repository at this point in the history
  • Loading branch information
wtomin committed Feb 11, 2025
1 parent b9c4e7e commit 6dea4aa
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion examples/hunyuanvideo/hyvideo/run_vae.py
Original file line number Diff line number Diff line change
Expand Up @@ -165,6 +165,7 @@ def main(args):
jit_level=args.jit_level,
jit_syntax_level=args.jit_syntax_level,
seed=args.seed,
distributed=args.use_parallel,
)

set_logger(name="", output_dir=args.output_path, rank=0)
Expand Down Expand Up @@ -265,7 +266,7 @@ def get_parser():
"If not provided, will search all videos under `real_video_dir` recursively.",
)
parser.add_argument(
"--video-column", dtype=str, default="video", help="The video column name in the provided Data file path."
"--video-column", type=str, default="video", help="The video column name in the provided Data file path."
)

# Other Group
Expand Down

0 comments on commit 6dea4aa

Please sign in to comment.