From c7b2635f01509e29aeff62bd06bffef41c3285fa Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Wed, 23 Apr 2025 20:37:48 +0300 Subject: [PATCH] Update model.py --- wanvideo/modules/model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py index 76e876c..834b289 100644 --- a/wanvideo/modules/model.py +++ b/wanvideo/modules/model.py @@ -1161,7 +1161,9 @@ class WanModel(ModelMixin, ConfigMixin): if self.teacache_use_coefficients: rescale_func = np.poly1d(self.teacache_coefficients[self.teacache_mode]) temb = e if self.teacache_mode == 'e' else e0 - accumulated_rel_l1_distance += rescale_func(((temb-previous_modulated_input).abs().mean() / previous_modulated_input.abs().mean()).cpu().item()) + accumulated_rel_l1_distance += rescale_func(( + (temb.to(device) - previous_modulated_input).abs().mean() / previous_modulated_input.abs().mean() + ).cpu().item()) else: temb_relative_l1 = relative_l1_distance(previous_modulated_input, e0) accumulated_rel_l1_distance = accumulated_rel_l1_distance.to(e0.device) + temb_relative_l1