xinyun/TeaCache
1
0
Fork 0
mirror of https://git.datalinker.icu/ali-vilab/TeaCache synced 2025-12-08 20:34:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update and rename teacache_lumina2_v1.py to teacache_lumina2.py

Browse Source
This commit is contained in:
spawner 2025-06-08 20:39:07 +08:00 committed by GitHub
parent 0a9b0358ca
commit ff6a083896
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
TeaCache4Lumina2/teacache_lumina2_v1.py → TeaCache4Lumina2/teacache_lumina2.py
View File

@ -68,7 +68,8 @@ def teacache_forward_working(
current_cache["accumulated_rel_l1_distance"] = 0.0
else:
if current_cache["previous_modulated_input"] is not None:
coefficients = [393.76566581, -603.50993606, 209.10239044, -23.00726601, 0.86377344] # taken from teacache_lumina_next.py
# v1 coefficients，you can switch it to [225.7042019806413, -608.8453716535591, 304.1869942338369, 124.21267720116742, -1.4089066892956552] as v2
coefficients = [393.76566581, -603.50993606, 209.10239044, -23.00726601, 0.86377344]
rescale_func = np.poly1d(coefficients)
prev_mod_input = current_cache["previous_modulated_input"]
prev_mean = prev_mod_input.abs().mean()