From d555ea056b2b76706a2d06a4e7bd947a2d0368e7 Mon Sep 17 00:00:00 2001 From: bubbliiiing <3323290568@qq.com> Date: Fri, 24 Jul 2026 16:59:48 +0800 Subject: [PATCH] Update speed test --- examples/wan2.1/predict_i2v.py | 2 +- examples/wan2.1/predict_i2v_speed.py | 2 +- examples/wan2.1/predict_i2v_speed.sh | 2 +- examples/wan2.1/predict_t2v.py | 2 +- examples/wan2.1/predict_t2v_speed.py | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/examples/wan2.1/predict_i2v.py b/examples/wan2.1/predict_i2v.py index 2f56563..72905c6 100755 --- a/examples/wan2.1/predict_i2v.py +++ b/examples/wan2.1/predict_i2v.py @@ -247,7 +247,7 @@ if coefficients is not None: coefficients, num_inference_steps, teacache_threshold, num_skip_start_steps=num_skip_start_steps, offload=teacache_offload ) -if cfg_skip_ratio is not None: +if cfg_skip_ratio is not None and cfg_skip_ratio > 0: print(f"Enable cfg_skip_ratio {cfg_skip_ratio}.") pipeline.transformer.enable_cfg_skip(cfg_skip_ratio, num_inference_steps) diff --git a/examples/wan2.1/predict_i2v_speed.py b/examples/wan2.1/predict_i2v_speed.py index d47fba3..8e69bc0 100644 --- a/examples/wan2.1/predict_i2v_speed.py +++ b/examples/wan2.1/predict_i2v_speed.py @@ -23,7 +23,7 @@ from videox_fun.utils.fp8_optimization import (convert_model_weight_to_float8, r convert_weight_dtype_wrapper) from videox_fun.utils.lora_utils import merge_lora, unmerge_lora from videox_fun.utils.utils import (filter_kwargs, get_image_to_video_latent, - save_videos_grid) + save_videos_grid, timer_record) from videox_fun.utils.fm_solvers import FlowDPMSolverMultistepScheduler from videox_fun.utils.fm_solvers_unipc import FlowUniPCMultistepScheduler diff --git a/examples/wan2.1/predict_i2v_speed.sh b/examples/wan2.1/predict_i2v_speed.sh index 19d6760..d8f67be 100644 --- a/examples/wan2.1/predict_i2v_speed.sh +++ b/examples/wan2.1/predict_i2v_speed.sh @@ -5,7 +5,7 @@ export DIT_EXCEL_COL=0 VAE_EXCEL_COL=1 TOTAL_EXCEL_COL=2 # 14B 720P export DIT_EXCEL_ROW=1 VAE_EXCEL_ROW=1 TOTAL_EXCEL_ROW=1 python examples/wan2.1/predict_i2v_speed.py --model_name="models/Diffusion_Transformer/Wan2.1-I2V-14B-720P" \ - --GPU_memory_mode="model_full_load_and_qfloat8" --ulysses_degree=1 --ring_degree=1 --fsdp_text_encoder --fsdp_dit \ + --GPU_memory_mode="model_full_load_and_qfloat8" --ulysses_degree=1 --ring_degree=1 --fsdp_text_encoder --compile_dit \ --enable_teacache --teacache_threshold=0.30 --num_skip_start_steps=2 --cfg_skip_ratio=0.25 --shift=5 \ --sample_size 720 1280 --num_inference_steps=40 diff --git a/examples/wan2.1/predict_t2v.py b/examples/wan2.1/predict_t2v.py index c50db5f..757ba48 100755 --- a/examples/wan2.1/predict_t2v.py +++ b/examples/wan2.1/predict_t2v.py @@ -234,7 +234,7 @@ if coefficients is not None: coefficients, num_inference_steps, teacache_threshold, num_skip_start_steps=num_skip_start_steps, offload=teacache_offload ) -if cfg_skip_ratio is not None: +if cfg_skip_ratio is not None and cfg_skip_ratio > 0: print(f"Enable cfg_skip_ratio {cfg_skip_ratio}.") pipeline.transformer.enable_cfg_skip(cfg_skip_ratio, num_inference_steps) diff --git a/examples/wan2.1/predict_t2v_speed.py b/examples/wan2.1/predict_t2v_speed.py index 1b7c2cd..f6a78ce 100755 --- a/examples/wan2.1/predict_t2v_speed.py +++ b/examples/wan2.1/predict_t2v_speed.py @@ -246,7 +246,7 @@ for i in range(2): coefficients, num_inference_steps, teacache_threshold, num_skip_start_steps=num_skip_start_steps, offload=teacache_offload ) - if cfg_skip_ratio is not None: + if cfg_skip_ratio is not None and cfg_skip_ratio > 0: print(f"Enable cfg_skip_ratio {cfg_skip_ratio}.") pipeline.transformer.enable_cfg_skip(cfg_skip_ratio, num_inference_steps)