From 652b58f24c39ecc33584470699dae7152158aade Mon Sep 17 00:00:00 2001 From: hkz Date: Wed, 9 Apr 2025 15:24:08 +0800 Subject: [PATCH] fix teacache (#160) --- videox_fun/models/cache_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/videox_fun/models/cache_utils.py b/videox_fun/models/cache_utils.py index a2a327d..b9209fa 100644 --- a/videox_fun/models/cache_utils.py +++ b/videox_fun/models/cache_utils.py @@ -3,13 +3,13 @@ import torch def get_teacache_coefficients(model_name): - if "wan2.1-t2v-1.3b" or "wan2.1-fun-1.3b-inp" in model_name.lower(): + if "wan2.1-t2v-1.3b" or "wan2.1-fun-1.3b" in model_name.lower(): return [-5.21862437e+04, 9.23041404e+03, -5.28275948e+02, 1.36987616e+01, -4.99875664e-02] elif "wan2.1-t2v-14b" in model_name.lower(): return [-3.03318725e+05, 4.90537029e+04, -2.65530556e+03, 5.87365115e+01, -3.15583525e-01] elif "wan2.1-i2v-14b-480p" in model_name.lower(): return [2.57151496e+05, -3.54229917e+04, 1.40286849e+03, -1.35890334e+01, 1.32517977e-01] - elif "wan2.1-i2v-14b-720p" or "wan2.1-fun-14b-inp" in model_name.lower(): + elif "wan2.1-i2v-14b-720p" or "wan2.1-fun-14b" in model_name.lower(): return [8.10705460e+03, 2.13393892e+03, -3.72934672e+02, 1.66203073e+01, -4.17769401e-02] else: print(f"The model {model_name} is not supported by TeaCache.")