(model_name)
| 2 | import torch |
| 3 | |
| 4 | def get_teacache_coefficients(model_name): |
| 5 | if "wan2.1-t2v-1.3b" in model_name.lower() or "wan2.1-fun-1.3b" in model_name.lower() \ |
| 6 | or "wan2.1-fun-v1.1-1.3b" in model_name.lower() or "wan2.1-vace-1.3b" in model_name.lower(): |
| 7 | return [-5.21862437e+04, 9.23041404e+03, -5.28275948e+02, 1.36987616e+01, -4.99875664e-02] |
| 8 | elif "wan2.1-t2v-14b" in model_name.lower(): |
| 9 | return [-3.03318725e+05, 4.90537029e+04, -2.65530556e+03, 5.87365115e+01, -3.15583525e-01] |
| 10 | elif "wan2.1-i2v-14b-480p" in model_name.lower(): |
| 11 | return [2.57151496e+05, -3.54229917e+04, 1.40286849e+03, -1.35890334e+01, 1.32517977e-01] |
| 12 | elif "wan2.1-i2v-14b-720p" in model_name.lower() or "wan2.1-fun-14b" in model_name.lower() or "wan2.2-fun" in model_name.lower() \ |
| 13 | or "wan2.2-i2v-a14b" in model_name.lower() or "wan2.2-t2v-a14b" in model_name.lower() or "wan2.2-ti2v-5b" in model_name.lower() \ |
| 14 | or "wan2.2-s2v" in model_name.lower() or "wan2.1-vace-14b" in model_name.lower() or "wan2.2-vace-fun" in model_name.lower() \ |
| 15 | or "wan2.2-animate" in model_name.lower(): |
| 16 | return [8.10705460e+03, 2.13393892e+03, -3.72934672e+02, 1.66203073e+01, -4.17769401e-02] |
| 17 | elif "qwen-image" in model_name.lower(): |
| 18 | # Copied from https://github.com/chenpipi0807/ComfyUI-TeaCache/blob/main/nodes.py |
| 19 | return [-4.50000000e+02, 2.80000000e+02, -4.50000000e+01, 3.20000000e+00, -2.00000000e-02] |
| 20 | else: |
| 21 | print(f"The model {model_name} is not supported by TeaCache.") |
| 22 | return None |
| 23 | |
| 24 | |
| 25 | class TeaCache(): |
no outgoing calls
no test coverage detected