From 278a4c3cd29efd988f971fcecf62c4f31860beb6 Mon Sep 17 00:00:00 2001 From: Tianhao-Qi <37017794+Tianhao-Qi@users.noreply.github.com> Date: Thu, 17 Apr 2025 19:06:37 +0800 Subject: [PATCH] fix bugs in if_else conditions for function get_teacache_coefficients (#173) --- 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 b9209fa..871b10c 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" in model_name.lower(): + if "wan2.1-t2v-1.3b" in model_name.lower() 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" in model_name.lower(): + elif "wan2.1-i2v-14b-720p" in model_name.lower() 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.")