From 16f978b81dd575e9899332e2de70ed2dde94d8aa Mon Sep 17 00:00:00 2001 From: Uxito-Ada <414416158@qq.com> Date: Tue, 14 Jul 2026 08:29:33 +0000 Subject: [PATCH] Fix Wan2.1-T2V-14B H100 example --- examples/wan_video/README.md | 8 +++++++- examples/wan_video/wan21_14b_text_to_video_h100.py | 14 +++++++------- 2 files changed, 14 insertions(+), 8 deletions(-) diff --git a/examples/wan_video/README.md b/examples/wan_video/README.md index e879a9a..c97a1aa 100644 --- a/examples/wan_video/README.md +++ b/examples/wan_video/README.md @@ -203,6 +203,12 @@ python examples/wan_video/wan21_14b_text_to_video_h100.py --gpu_num 2 --prompt " python examples/wan_video/wan21_14b_text_to_video_h100.py --resolution 720p --aspect_ratio 16:9 ``` +**Hint:** +If you encounter error like `RuntimeError: unable to open shared memory object`, `OSError: Too many open files`, solve it with: +```bash +ulimit -n 65535 +``` + **Features:** - 14B parameter model for high-quality generation - CFG parallel enabled (cfg_scale=5.0) @@ -574,4 +580,4 @@ python examples/wan_video/async_wan22_14b_image_to_video_distill_h100.py \ - FP8 quantization reduces memory by ~50% - Ray distributed inference enables efficient multi-GPU scaling - Feature cache is configured via `ModelRuntimeConfig.feature_cache_config` during pipeline initialization -- Parallel config is automatically set based on cfg_scale values \ No newline at end of file +- Parallel config is automatically set based on cfg_scale values diff --git a/examples/wan_video/wan21_14b_text_to_video_h100.py b/examples/wan_video/wan21_14b_text_to_video_h100.py index 68bdcdb..d964ce5 100644 --- a/examples/wan_video/wan21_14b_text_to_video_h100.py +++ b/examples/wan_video/wan21_14b_text_to_video_h100.py @@ -44,13 +44,12 @@ def get_dit_path_list(model_root): """Generate DiT model paths based on model_root.""" return [ - f"{model_root}/diffusion_pytorch_model-00001-of-00007.safetensors", - f"{model_root}/diffusion_pytorch_model-00002-of-00007.safetensors", - f"{model_root}/diffusion_pytorch_model-00003-of-00007.safetensors", - f"{model_root}/diffusion_pytorch_model-00004-of-00007.safetensors", - f"{model_root}/diffusion_pytorch_model-00005-of-00007.safetensors", - f"{model_root}/diffusion_pytorch_model-00006-of-00007.safetensors", - f"{model_root}/diffusion_pytorch_model-00007-of-00007.safetensors", + f"{model_root}/diffusion_pytorch_model-00001-of-00006.safetensors", + f"{model_root}/diffusion_pytorch_model-00002-of-00006.safetensors", + f"{model_root}/diffusion_pytorch_model-00003-of-00006.safetensors", + f"{model_root}/diffusion_pytorch_model-00004-of-00006.safetensors", + f"{model_root}/diffusion_pytorch_model-00005-of-00006.safetensors", + f"{model_root}/diffusion_pytorch_model-00006-of-00006.safetensors", ] @@ -99,6 +98,7 @@ def get_pipeline(parallelism=1, model_root=PPL_CONFIG["model_root"]): pipe_config.dit_config.parallel_config.device_ids = list(range(parallelism)) pipe_config.enable_denoising_parallel = True + pipe_config.dit_config.parallel_config.timeout = 3600 pipe.init(module_manager, pipe_config) return pipe