From dede1988e6b1354f52bf3b569430259c2e3af31e Mon Sep 17 00:00:00 2001 From: Artsiom Siamashka Date: Fri, 3 Apr 2026 20:12:31 +0200 Subject: [PATCH] Run Hunyuan's generate script with torchrun utility --- src/generate_videos.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/generate_videos.py b/src/generate_videos.py index 6da195f..f82d0df 100644 --- a/src/generate_videos.py +++ b/src/generate_videos.py @@ -14,6 +14,7 @@ import sys from pathlib import Path import torch +from torch.distributed.run import main as torch_run from src.logging_config import configure_logging, debug_log_lifecycle @@ -48,17 +49,13 @@ def _run_hunyuan_generate_in_process( argv: list[str], env_update: dict[str, str], ) -> None: - generate_script = hunyuan_dir / "generate.py" - if not generate_script.exists(): - raise FileNotFoundError(f"Hunyuan generate script not found at {generate_script}") - old_argv = sys.argv[:] old_cwd = Path.cwd() try: with _temporary_environ(env_update): os.chdir(hunyuan_dir) - sys.argv = ["generate.py", *argv] - runpy.run_path(str(generate_script), run_name="__main__") + sys.argv = argv + torch_run() finally: sys.argv = old_argv os.chdir(old_cwd) @@ -159,6 +156,8 @@ def main(argv: list[str] | None = None) -> int: LOGGER.info("GPU cache cleared") run_argv = [ + "--nproc_per_node=1", + "generate.py", "--prompt", prompt, "--image_path",