From f77c0dc5d8cdec3e22782b63740f7721055bc65f Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Wed, 26 Mar 2025 01:14:16 +0200 Subject: [PATCH] Update model_optimization_nodes.py --- nodes/model_optimization_nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes/model_optimization_nodes.py b/nodes/model_optimization_nodes.py index 2cf00e6..cc193bc 100644 --- a/nodes/model_optimization_nodes.py +++ b/nodes/model_optimization_nodes.py @@ -1204,7 +1204,7 @@ class CFGZeroStar: dot_product = torch.sum(positive_flat * negative_flat, dim=1, keepdim=True) squared_norm = torch.sum(negative_flat ** 2, dim=1, keepdim=True) + 1e-8 alpha = dot_product / squared_norm - alpha = alpha.view(batch_size, 1, 1, 1, 1) + alpha = alpha.view(batch_size, *([1] * (len(cond.shape) - 1))) matched_step_index = (sigmas == timestep[0] ).nonzero() if len(matched_step_index) > 0: