From 413f831ced1165057ee79be751f4fe80223588c1 Mon Sep 17 00:00:00 2001
From: bubbliiiing <3323290568@qq.com>
Date: Thu, 31 Jul 2025 19:28:17 +0800
Subject: [PATCH] Update Wan2.2 Speed

---
 examples/wan2.2/predict_t2v_speed.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/wan2.2/predict_t2v_speed.py b/examples/wan2.2/predict_t2v_speed.py
index b507566..6c6940c 100755
--- a/examples/wan2.2/predict_t2v_speed.py
+++ b/examples/wan2.2/predict_t2v_speed.py
@@ -281,7 +281,7 @@ elif GPU_memory_mode == "model_full_load_and_qfloat8":
 else:
     pipeline.to(device=device)
 
-while 1:
+for i in range(2):
     coefficients = get_teacache_coefficients(model_name) if enable_teacache else None
     if coefficients is not None:
         print(f"Enable TeaCache with threshold {teacache_threshold} and skip the first {num_skip_start_steps} steps.")