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 teacache_lumina2.py

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

2
TeaCache4Lumina2/teacache_lumina2.py
View File

@ -68,7 +68,7 @@ def teacache_forward_working(
current_cache["accumulated_rel_l1_distance"] = 0.0 current_cache["accumulated_rel_l1_distance"] = 0.0
else: else:
if current_cache["previous_modulated_input"] is not None: 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 coefficients = [225.7042019806413, -608.8453716535591, 304.1869942338369, 124.21267720116742, -1.4089066892956552]
rescale_func = np.poly1d(coefficients) rescale_func = np.poly1d(coefficients)
prev_mod_input = current_cache["previous_modulated_input"] prev_mod_input = current_cache["previous_modulated_input"]
prev_mean = prev_mod_input.abs().mean() prev_mean = prev_mod_input.abs().mean()